使用 python 向数据框添加一列（减法）

Question

我尝试在包含值 0 的数据框 df2 中添加一列 if(df2['P_ACT_KW'] - df2['P_SOUSCR']) < 0 else df2['P_ACT_KW']- df2['P_SOUSCR']。

if (df2['P_ACT_KW'] - df2['P_SOUSCR']) <0:
    df2['depassement']=0
else:
    df2['depassement']= (df2['P_ACT_KW'] - df2['P_SOUSCR'])

我收到此错误消息：

ValueError Traceback (most recent call last) in () ----> 1 if (df2['P_ACT_KW'] - df2['P_SOUSCR']) <0: 2 df2['depassement']=0 3 else: 4 df2['depassement']= (df2['P_ACT_KW'] - df2['P_SOUSCR'])

C:\Users\Demonstrator\Anaconda3\lib\site-packages\pandas\core\generic.py in nonzero(self) 890 raise ValueError("The truth value of a {0} is ambiguous. " 891 "Use a.empty, a.bool(), a.item(), a.any() or a.all()." --> 892 .format(self.class.name)) 893 894 bool = nonzero

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

有什么想法吗？

谢谢

Answer 1

您需要做的：

df2['depassement'] = np.where((df2['P_ACT_KW'] - df2['P_SOUSCR']) < 0), 0, df2['P_ACT_KW'] - df2['P_SOUSCR'])

if 不明白如何比较类似数组的结构，因此出现错误，这里我们可以使用 np.where 来比较所有行以生成掩码，条件为真时设置为 0 否则执行减法

Answer 2

IIUC:

df2['depassement'] = df2['P_ACT_KW'] - df2['P_SOUSCR']
df2[df2['depassement'] < 0, 'depassement'] = 0

这也应该有效：

df2['depassement'] = df2.P_ACT_KW.sub(df2.P_SOUSCR).apply(lambda x: max(x, 0))

使用 python 向数据框添加一列（减法）

add one column (subtraction) to a dataframe with python

python

calculated-columns

dataframe

pandas