基于另一个多个数据框列的新列

Question

下面是我的数据框。 |一个 |乙 | |---|---| | 12| 0 | | 1 | 21| | 0 | 0 |

现在想添加一个列名“c”，当 a 或 b 为非零时返回 yes，当 a 和 b 为零时返回 no

a	b	c
12	0	yes
1	21	yes
0	0	no

Answer 1

如果需要通过0 测试所有列比较，并通过DataFrame.all 测试每行的所有值是否设置yes，numpy.where '否'：

df['c'] = np.where(df.eq(0).all(axis=1), 'no','yes')

print (df)
    a   b    c
0  12   0  yes
1   1  21  yes
2   0   0   no

另一个想法：

df['c'] = df.ne(0).any(axis=1).map({False: 'no',True:'yes'})

如果可能有多个列并且只需要测试a,b 列：

cols = ['a','b']
df['c'] = np.where(df[cols].eq(0).all(axis=1), 'no','yes')