python - 将 DataFrame 列表保存到多表 Excel 电子表格

Question

如何将 DataFrame 列表导出到一个 Excel 电子表格中？
to_excel 的文档状态:

Notes
If passing an existing ExcelWriter object, then the sheet will be added to the existing workbook. This can be used to save different DataFrames to one workbook

writer = ExcelWriter('output.xlsx')
df1.to_excel(writer, 'sheet1')
df2.to_excel(writer, 'sheet2')
writer.save()

在此之后，我想我可以编写一个函数，将 DataFrame 列表保存到一个电子表格中，如下所示:

from openpyxl.writer.excel import ExcelWriter
def save_xls(list_dfs, xls_path):
    writer = ExcelWriter(xls_path)
    for n, df in enumerate(list_dfs):
        df.to_excel(writer,'sheet%s' % n)
    writer.save()

但是(有两个小的 DataFrames 列表，每个 DataFrames 可以单独保存 to_excel)，引发了异常(编辑:traceback 已删除):

AttributeError: 'str' object has no attribute 'worksheets'

大概我没有调用 ExcelWriter正确，我应该怎么做才能做到这一点？

Answer 1

你应该使用 pandas 自己的 ExcelWriter 类:

from pandas import ExcelWriter
# from pandas.io.parsers import ExcelWriter

然后save_xls函数按预期工作:

def save_xls(list_dfs, xls_path):
    with ExcelWriter(xls_path) as writer:
        for n, df in enumerate(list_dfs):
            df.to_excel(writer,'sheet%s' % n)