python - 如何为数据框中的特定类找到使用熊猫的意思？

Question

我想知道是否有一种直接的方法可以只找到数据框中特定类的平均值。我正在使用的文件用于对字母进行分类，因此它看起来像这样：

    class     feature 1    feature 2    feature 3   feature 4 
0   A         2            3            4           1
1   A         3            1            1           0
2   B         6            7            8           8

有没有一种简单的方法我只能找到每个特征的“A”的平均值？

标签： pythonpandas

Answer 1

您可以使用它来获取具有值的class列中的行的平均值A

df[df['class']=='A'].groupby('class')[list(df)[1:]].mean()

输出

         feature 1  feature 2  feature 3  feature 4
feature                                                   
A              2.5        2.0        2.5        0.5

这将为mean4feature n列提供 。