按列位置分组的熊猫数据框

Question

我有一个函数可以对熊猫数据框进行分组。问题是我的数据框可以有可变数量的列。我想聚合：将最后一列与第一列相加。最后一列的名称不同，但第一列的名称是固定的。

我怎样才能达到这个组？我尝试使用 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