python -- 解决使用groupby函数之后索引上移问题

【问题描述】

对于一个数据框,当我们对按每一个变量进行groupby之后再对另一个指标进行分组统计时,得到的的结果转成DataFrame之后会发现,统计的指标标签会上移。这回造成索引问题。

例子:

test_0 = {"id":[1,1,2,3,3,4,5,5],"price":[5,6,8,3,4,6,9,5],"amount":[1,1,2,1,1,1,2,1],"status":['sale','sale','no','no','sale','no','sale','no']}
test = pd.DataFrame(test_0)
test = test[['id','price','amount','status']]
test

test_1 = pd.DataFrame(test.groupby('id')['amount'].sum())
test_1


【解决办法】

test_2 = pd.DataFrame(test.groupby('id',as_index=False)['amount'].sum())

问题解决!

(仅供个人学习,不负责任~~~~~~~~~~~)


猜你喜欢

转载自blog.csdn.net/august1226/article/details/80667398