查找数据框中每列的平均值，按标签分组，不包括一个标签

Posted 2021-04-13

我有一个如下所示的数据帧：

    CPU Memory Disk  Label
0    21     28   29      0
1    46     53   55      1
2    48     45   49      2
3    48     52   50      3
4    51     54   55      4
5    45     50   56      5
6    50     83   44     -1 

我想要的是groupby并找到每个标签的平均值。到目前为止，我有这个dataset.groupby('Label')['CPU', 'Memory', 'Disk'].mean()工作得很好，得到的结果如下：

Label           CPU     Memory       Disk 
    -1     46.441176  53.882353  54.176471
     0     48.500000  58.500000  60.750000
     1     45.000000  51.000000  60.000000
     2     54.000000  49.000000  56.000000
     3     55.000000  71.500000  67.500000
     4     53.000000  70.000000  71.000000
     5     21.333333  30.000000  30.666667

我唯一没有找到的是如何排除标记为-1的所有内容。有没有办法做到这一点？

答案

您可以在分组之前过滤数据帧：

# Exclude rows with Label=-1
dataset = dataset.loc[dataset['Label'] != -1]
# Group by on filtered result
dataset.groupby('Label')['CPU', 'Memory', 'Disk'].mean()