如果其他单元格包含“某物”，如何在 2 个单元格中设置值

Question

我有一些熊猫数据框：

a,b,c
AAA,,
DDD,,
KKK,,
AAA,,  

我想在“A”列中搜索，如果“A”列中的字符串包含单词“AAA”，我需要在“B”列中设置值“BBB”，在“C”列中设置值“CCC”。
所以，我想得到如下结果：

a,b,c
AAA,BBB,CCC
DDD,,
KKK,,
AAA,BBB,CCC

我用numpy写了代码：

df['b'] = pd.np.where(df.a.str.contains("AAA"), "BBB", '')

如何扩展它以使用 'b' 和 'c' 列？

Answer 1

你可以使用双np.where:

mask = df.a.str.contains("AAA")
df['b'] = pd.np.where(mask, "BBB", '')
df['c'] = pd.np.where(mask, "CCC", '')

或assign:

mask = df.a.str.contains("AAA")
df = df.assign(b=pd.np.where(mask, "BBB", ''), c=pd.np.where(mask, "CCC", ''))

如果需要使用一个np.where 创建多个列，则需要创建Nx1 掩码：

mask = df.a.str.contains("AAA")[:, None]
df[['b','c']] = np.where(mask, ['BBB','CCC'], ['',''])
print (df)
     a    b    c
0  AAA  BBB  CCC
1  DDD          
2  KKK          
3  AAA  BBB  CCC