一列的平均值，基于另一列的相等条件

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.0606的值。也许还有其他代码？我得到了我想要得到的价值，但是有更好的方法吗？你知道吗

Answer 1

改用这个：

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

或者

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