Pandas: fill in NaN values with dictionary references another column

Question

I have a dictionary that looks like this

dict = {'b' : '5', 'c' : '4'}

My dataframe looks something like this

   A  B
0  a  2
1  b  NaN
2  c  NaN

Is there a way to fill in the NaN values using the dictionary mapping from columns A to B while keeping the rest of the column values?

Answer 1

You can map dict values inside fillna

df.B = df.B.fillna(df.A.map(dict))

print(df)

    A   B
0   a   2
1   b   5
2   c   4

Answer 2

This can be done simply

df['B'] = df['B'].fillna(df['A'].apply(lambda x: dict.get(x)))

This can work effectively for a bigger dataset as well.