PHP前端开发

如何使用 groupby() 函数计算 DataFrame 中按特定列分组数据的平均值?

百变鹏仔 5天前 #Python
文章标签 平均值

groupby() 函数应用查询

为了查询给定 dataframe 中按特定列分组数据的平均值,可以使用 groupby() 函数。

df = pd.dataframe(    {        "major": ["math", "physics", "math", "math", "physics", "math", "math", "physics", "demography"],        "grade": ["3", "2", "2", "1", "1", "3", "3", "2", "2"],        "age": [20, 18, 19, 21, 20, 19, 19, 18, 20],    })

要输出按 "major" 列分组、计算 "grade" 列均值的表格,可以使用以下代码:

df.groupby(["major"]).mean("grade")

输出结果为:

Grade                1           2           3MajorDemography<br>Math<br>Pyhsics

其中,表格中的值为各组 "grade" 字段的均值。