当使用np.where()函数替换包含NAN的字符串值时出现以下问题:
A
0 a
1 NAN
2 a,b
3 b
我想创建一个新的列B,以指示列A中是否包含“ a”为1:
A B
0 a 1
1 NAN 0
2 a,b 1
3 b 0
但是当我使用时:
df['B'] = np.where(df['A'].str.contains('a',case=False),1, 0)
结果还将NAN行也转换为1,不确定我的代码有什么问题:
A B
0 a 1
1 NAN 1
2 a,b 1
3 b 0