熊猫数据框按列位置分组

Question

我有一个函数可以在 pandas 数据帧上进行分组。问题是我的数据框可以有可变数量的列。我想汇总：将最后一列与第一列相加。最后一列的名称不同，但第一列的名称是固定的。

我怎样才能实现分组？我尝试使用 iloc 并使用 df.columns[-1] 获取最后一列的列名，但是，这些技巧似乎都不起作用。

有没有比将最后一列名称更改为某个通用值更好的方法来实现这一点？

Answer 1

df.groupby(df.columns[0])[df.columns[-1]].sum() 应该可以工作。

示例：

df = pd.DataFrame({
    'a': [1,1,2,2],
    'b': [1,2,3,4]
})

df.groupby(df.columns[0])[df.columns[-1]].sum()
#a
#1    3
#2    7
#Name: b, dtype: int64

Answer 2

只需使用iloc选择的Series，@Psidom借用的数据：

s = df.iloc[:, -1].groupby(df.iloc[:, 0]).sum()
print (s)
a
1    3
2    7
Name: b, dtype: int64