Pandas按groupby求和，但不包括某些列

Question

在Pandas数据帧上执行groupby的最佳方法是什么，但要从该groupby中排除某些列？e、 g.我有以下数据框：

Code   Country      Item_Code   Item    Ele_Code    Unit    Y1961    Y1962   Y1963
2      Afghanistan  15          Wheat   5312        Ha      10       20      30
2      Afghanistan  25          Maize   5312        Ha      10       20      30
4      Angola       15          Wheat   7312        Ha      30       40      50
4      Angola       25          Maize   7312        Ha      30       40      50

我想按列Country和Item_Code分组，只计算列Y1961、Y1962和Y1963下的行的总和。生成的数据帧应如下所示：

Code   Country      Item_Code   Item    Ele_Code    Unit    Y1961    Y1962   Y1963
2      Afghanistan  15          C3      5312        Ha      20       40       60
4      Angola       25          C4      7312        Ha      60       80      100

现在我要做的是：

df.groupby('Country').sum()

但是，这也会将Item_Code列中的值相加。有没有什么方法可以指定在sum()操作中要包括哪些列以及要排除哪些列？

Answer 1

如果您正在寻找一种更通用的方法来应用于许多列，那么您可以做的是构建列名列表并将其作为分组数据帧的索引传递。在您的情况下，例如：

columns = ['Y'+str(i) for year in range(1967, 2011)]

df.groupby('Country')[columns].agg('sum')

Answer 2

您可以选择groupby的列：

In [11]: df.groupby(['Country', 'Item_Code'])[["Y1961", "Y1962", "Y1963"]].sum()
Out[11]:
                       Y1961  Y1962  Y1963
Country     Item_Code
Afghanistan 15            10     20     30
            25            10     20     30
Angola      15            30     40     50
            25            30     40     50

请注意，传递的列表必须是列的子集，否则您将看到KeyError。

Answer 3

agg函数将为您执行此操作。将列和函数作为带列的dict传递，输出：

df.groupby(['Country', 'Item_Code']).agg({'Y1961': np.sum, 'Y1962': [np.sum, np.mean]})  # Added example for two output columns from a single input column

这将仅显示“按列分组”和指定的聚合列。在本例中，我包含了两个应用于“Y1962”的agg函数。

要获得您希望看到的内容，请将“分组依据”中的其他列包括在内，并对框架中的Y变量应用求和：

df.groupby(['Code', 'Country', 'Item_Code', 'Item', 'Ele_Code', 'Unit']).agg({'Y1961': np.sum, 'Y1962': np.sum, 'Y1963': np.sum})