如果添加列，请注意序列的真值不明确

Question

我想在dataframe a中添加一列，

a = pd.DataFrame([[1,2],[3,4]],columns=['A','B'])
if a['B'] > a['A']:
    a['C']='是'
else:
    a['C']='否'

ValueError：序列的真值不明确。使用a.empty、a.bool（）、a.item（）、a.any（）或a.all（）。

Answer 1

还有一个选择是

np.choose()

它是在

np.choose('是','否')

Answer 2

是，其中或numpy。选择：

a = pd.DataFrame([[2,1],[3,4]],columns=['A','B'])
a['C'] = np.select([a['B']>a['A']], '是', default = '否')
print(a)

返回：

   A  B  C
0  2  1  否
1  3  4  是

它很容易扩展到更多的条件：

a = pd.DataFrame([[2,1],[3,4],[1,10]],columns=['A','B'])

condlist = [
    a['B'] > 5*a['A'],
    a['B'] > a['A']
]
valuelist = ['是', '否']

a['C'] = np.select(condlist, valuelist, default=np.nan)
print(a)

返回：

   A   B    C
0  2   1   nan
1  3   4    否
2  1  10    是

Answer 3

使用numpy.where：

#swapped 2,1
a = pd.DataFrame([[2,1],[3,4]],columns=['A','B'])
a['C'] = np.where(a['B']>a['A'], '是','否')
print (a)
   A  B  C
0  2  1  否
1  3  4  是

代码的问题在于是否使用：

print (a['B']>a['A'])
0    False
1     True
dtype: bool

如果无法决定要做什么，则返回布尔掩码和。

还可以使用熊猫的if真理陈述进行检查。