一列的平均值，基於另一列的相等條件

Question

如果列'pgain'== 4，我想要列'vgain'的平均值。vgain和pgain均為int64類型，沒有缺失值。

我嘗試了這個：

df['vgain'].groupby(df['pgain'] == 4).mean()

這給了我以下輸出：

pgain
False    2.851485
True     2.060606
Name: vgain, dtype: float64

有沒有一種方法可以直接顯示僅2.060606的值。 也許還有其他一些代碼？ 我獲得了想要獲得的價值，但是有更好的方法嗎？

Answer 1

使用此代替：

df.query('pgain == 4')['vgain'].mean()

要么

df.loc[df.pgain == 4,'vgain'].mean()