python - 如何通过 Python 获取特定类别的平均值

Question

我想知道如何通过 Python 计算特定类别的平均值？我有一个名为 demo.csv 的 csv 文件

 import pandas as pd
 import numpy as np

#loading the data into data frame
X = pd.read_csv('demo.csv')

感兴趣的两列是 Category 和 Totals 列:

Category Totals estimates
2   2777    0.43
4   1003    0.26
4   3473    0.65
4   2638    0.17
1   2855    0.74
0   2196    0.13
0   2630    0.91
2   2714    0.39
3   2472    0.51
0   1090    0.12

我有兴趣找到与 Category 2 相对应的总计的平均值。我知道如何在 excel 上执行此操作，我只是过滤以仅显示类别 2 并获得平均值(最终为 2745.5)，但我如何通过 Python 对此进行编码？

Answer 1

您可以将数据框限制为所需行的子集 (Category=2)，然后取对应于 Totals 列的列的平均值，如下所示:

df[df['Category'] == 2]['Totals'].mean()
2745.5