在給定特定條件的情況下，如何在 pandas dataframe 上使用 .count()？

Question

我已經嘗試了這兩個：

    df["Buys"] = df.groupby(['symbol','month','counterparty','side'])[df[df['side']=='Buy']].count()
    df["Sells"] = df.groupby(['symbol','month','counterparty','side'])[df['side']=='Sell'].count()

但是第一次我得到錯誤“DataFrame 不可哈希”，第二次“True 不是一個鍵”。 我在這里做錯了什么？

Answer 1

好吧，我想我明白了一半。 我做了

dfBuys = df.groupby(['symbol','month'])['side'].apply(lambda x: (x=='Buy').sum()).reset_index(name='count')
df["Buys"] = dfBuys["count"]

dfBuys["count"] 正是我要找的，但無論出於何種原因，有時 df["Buys"] 不會填充，但在大多數情況下它都有效。