python-3.x - 如何从 groupby() 中取消堆叠两列

Question

我有下面的代码来计算每个销售人员在不同地点销售的产品数量

df = df.groupby(['location','sales person','item']).size().unstack().fillna(0) 
df.to_excel(r'c:\users\Tom\Project\info.xlsx')

我的输出如下，但我需要填充 A 列中的空单元格，这意味着我想在第 3 行看到“东”，在第 5 行看到“西”。有没有办法做到这一点？谢谢

Answer 1

我想你想要的reset_index方法：

df.reset_index().to_excel(r'c:\users\Tom\Project\info.xlsx')

我显然没有你的数据，但这里有一个类似的例子：

df = pd.read_csv('https://raw.githubusercontent.com/mwaskom/seaborn-data/master/iris.csv')
d = df.groupby(['species','petal_length'])['sepal_length'].describe()
d
d.reset_index()

Answer 2

你可以试试：

df.reset_index().to_excel(r'c:\users\Tom\Project\info.xlsx', index=False)