我需要使用 pandas dataframe 根据第二个分类变量中的值来估算分类变量的缺失值

Question

Impute the missing values in first categorical variable(values- [0,1]) in such a way that if the values in second categorical variable(values- [1, 2, 3]) is either 2 or 3 then set the missing value in that row for first column to be 1 else set it to 0.

problem-

Col A	Col B
0	1
1	2
NaN	3
NaN	2
NaN	1
0	1

Expected-

Col A	Col B
0	1
1	2
1	3
1	2
0	1
0	1

Answer 1

Use Series.fillna for replace missing values by 1 if 2,3 tested by Series.isin and converted to integers for True, False to 1,0 mapping:

df['Col A'] = df['Col A'].fillna(df['Col B'].isin([2,3]).astype(int))