数据帧真假值

Question

我现在有这个可以用的

df['AStrategy'] = df['Areturn'][df['BUY'] == True]
df['BStrategy'] = df['Breturn'][df['SELL'] == True]

我想添加一个条件，在这里我只会出售时，购买是虚假的如下，但它没有工作

df['AStrategy'] = df['Areturn'][df['BUY'] == True]
df['BStrategy'] = df['Breturn'][df['SELL'] == True & df['BUY'] == False]

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

为什么它不工作？

有什么建议吗？非常感谢。

Answer 1

如果您只检查真/假，您只需执行以下操作：

df['BStrategy'] = df['Breturn'][df['SELL'] & ~df['BUY']]

由Python的操作员优先级解释。

因为：=比更不合适，所以您的代码不起作用

我给您的解决方案之所以有效，是因为：~x的优先级高于

import pandas as pd
import numpy as np

np.random.seed(1234)

df = pd.DataFrame({
    'SELL':np.random.choice([True,False],6),
    'BUY':np.random.choice([True,False],6),
    'Breturn':np.random.randint(10,size=6),
})

df['BStrategy'] = df['Breturn'][df['SELL'] & ~df['BUY']]

print(df)

返回：

     BUY  Breturn   SELL  BStrategy
0   True        6  False        NaN
1  False        8  False        NaN
2  False        0   True        0.0
3  False        5  False        NaN
4  False        0   True        0.0
5  False        9   True        9.0

Answer 2

df['BStrategy'] = df['Breturn'][(df['SELL'] == True) & (df['BUY'] == False)]

它应该是有效的，要应用多个过滤器，您需要括号