python - 使用 Python Pandas 在一个 Excel 文件中创建多个电子表格

Question

如标题所示，我需要使用 Pandas 将多个电子表格创建到一个 excel 文件中。同时this thread和 this one所有提供的解决方案，我认为我的情况有点不同。这两种情况都使用类似于:

writer = pd.ExcelWriter('output.xlsx')
DF1.to_excel(writer,'Sheet1')
DF2.to_excel(writer,'Sheet2')
writer.save()

但是，问题是我不能同时在内存中保存多个数据帧，因为它们中的每一个都太大了。我的数据可以是这个的复杂版本:df = pd.DataFrame(dict(A=list('aabb'), B=range(4), C=range(6,10)))
输出:A B C
0 一 0 6
1个1个7
2乙2 8
3 乙 3 9


我打算使用 grplist 中的项目 ['a', 'b', 'c'] 来执行某种计算并最终生成单独的电子表格data['A'] == a 通过 c :

data = pd.read_csv(fileloc)
grplist = [['a','b','c'],['d','e','f']]
for groups, numbers in zip(grplist, range(1, 5)):
    for category in groups:
        clean = data[(data['A'] == category) & (data['B'] == numbers)]['C']
        # --------My calculation to generate a dataframe--------
        my_result_df = pd.DataFrame(my_result)
        writer = ExcelWriter('my_path_of_excel')
        my_resultdf.to_excel(writer, 'Group%s_%s' % (numbers, category[:4]))
        writer.save()
        gc.collect()

遗憾的是我的代码没有创建多个电子表格，因为 groups, numbers 是循环的。我只能在我的 excel 中的单个电子表格中获得最后的结果。我能做什么？

这是我在这里的第一篇文章。我希望我遵守每条规则，这样这个话题就能顺利结束。如果有任何需要修改或改进的地方，请告诉我。感谢您的帮助:)

Answer 1

考虑df

df = pd.DataFrame(dict(A=list('aabb'), B=range(4)))

遍历组并打印

for name, group in df.groupby('A'):
    print('{}\n\n{}\n\n'.format(name, group))

a

   A  B
0  a  0
1  a  1


b

   A  B
2  b  2
3  b  3

---

to_excel

writer = pd.ExcelWriter('output.xlsx')
for name, group in df.groupby('A'):
    group.to_excel(writer, name)
writer.save()
writer.close()