Pandas 2个累积列的累积和

Question

尝试将累积列添加到 pandas df。

已尝试此代码，但得到一个 NaN：

df['Total_Coins_Bought'] = df.query("side == 'buy'")['amount'].cumsum()
df['Total_Coins_Sold'] = -df.query("side == 'sell'")['amount'].cumsum()

df['Total_Coins'] = df['Total_Coins_Bought']-df['Total_Coins_Sold'] 

只希望构建 Total_Coins 列； 2 个字段 Total_Coins_Bought 和 Total_Coins_Sold 是我只是想一步一步地构建这个

Answer 1

df["sign_amount"] = df["amount"] * df["side"].map({"buy": +1, "sell": -1})
df["total_coins"] = df["sign_amount"].cumsum()

基本上，我添加了一个sign_amount列，它与amount列相同，但如果side是“买入”则乘以+1，如果side是“sell”则乘以-1。 cumsum列的sign_amount应该是您想要的结果。