python - 如何根据先前的状态在 Pandas 数据框中创建列？

Question

我有一个带有时间序列索引的 pandas Dataframe。一列包含买入信号，另一列包含卖出信号。

buy     0 0 1 0 1 1 0 1 1 1 0 1 0 0 0 0 0 0 0 0 
sell    0 0 0 0 0 0 0 0 0 0 0 0 0 0 1 1 0 1 1 0

我想给这个简单的模型添加一个状态:任何时候最多有1只股票:只有当你目前什么都没有时才买入，当你有股票时才卖出。

buy     0 0 1 0 1 1 0 1 1 1 0 1 0 0 0 0 0 0 0 0 
sell    0 0 0 0 0 0 0 0 0 0 0 0 0 0 1 1 0 1 1 0
wallet  0 0 1 1 1 1 1 1 1 1 1 1 1 1 0 0 0 0 0 0

我如何根据 df['buy'] 和 df['sell'] 创建 df['wallet'] 列>?

Answer 1

这不是最漂亮的解决方案，但我相信它可以满足您的需求。

import pandas as pd

df = pd.DataFrame({ 'buy':  [0,0,0,1,0,1,1,1,1,1,0,0,0,0,0,0], 
                    'sell': [0,0,0,0,0,0,0,0,0,0,0,0,0,1,1,1]   })


df['wallet'] = pd.np.where(df['buy'] - df['sell'] < 0, 0, df['buy'].cumsum().count() - df['sell'].cumsum())
df.loc[df['wallet'] > 1, 'wallet'] = 1