将多个 pandas 数据帧写入多个 excel 工作表

Question

我希望代码通过循环运行 12345，将其输入到工作表中，然后从 54321 开始并执行相同的操作，只是将数据帧输入到新工作表中，但在同一个工作表中工作簿。下面是我的代码。

workbook = xlsxwriter.Workbook('Renewals.xlsx')

groups = ['12345', '54321']

for x in groups:

    (Do a bunch of data manipulation and get pandas df called renewals)

    writer = pd.ExcelWriter('Renewals.xlsx', engine='xlsxwriter')
    worksheet = workbook.add_worksheet(str(x))
    renewals.to_excel(writer, sheet_name=str(x))

当这个运行时，我只剩下一个只有 1 个工作表 (54321) 的工作簿。

Answer 1

尝试这样的事情：

import pandas as pd
#initialze the excel writer
writer = pd.ExcelWriter('MyFile.xlsx', engine='xlsxwriter')

#store your dataframes in a  dict, where the key is the sheet name you want
frames = {'sheetName_1': dataframe1, 'sheetName_2': dataframe2,
        'sheetName_3': dataframe3}

#now loop thru and put each on a specific sheet
for sheet, frame in  frames.iteritems(): # .use .items for python 3.X
    frame.to_excel(writer, sheet_name = sheet)

#critical last step
writer.save()

Answer 2

import pandas as pd
writer = pd.ExcelWriter('Renewals.xlsx', engine='xlsxwriter')

renewals.to_excel(writer, sheet_name=groups[0])
renewals.to_excel(writer, sheet_name=groups[1])
writer.save()

Answer 3

根据已接受的答案，您会发现 sheet 名称包含无效字符或太长会导致保存失败的情况。如果您使用 sheet 名称的分组值作为示例，则可能会发生这种情况。辅助函数可以解决这个问题并减轻您的痛苦。

def clean_sheet_name(sheet):
"""Clean sheet name so that it is a valid Excel sheet name.

Removes characters in []:*?/\ and limits to 30 characters.

Args:
    sheet (str): Name to use for sheet.
    
Returns:
    cleaned_sheet (str): Cleaned sheet name.
"""
if sheet in (None, ''):
    return sheet
clean_sheet = sheet.translate({ord(i): None for i in '[]:*?/\'})
if len(clean_sheet) > 30: # Set value you feel is appropriate
    clean_sheet = clean_sheet[:30]
return clean_sheet

然后在写入Excel之前添加对辅助函数的调用。

for sheet, frame in groups.items():
    # Clean sheet name for length and invalid characters
    sheet = clean_sheet_name(sheet)
    frame.to_excel(writer, sheet_name = sheet, index=False)
writer.save()

将多个 pandas 数据帧写入多个 excel 工作表

Writing multiple pandas dataframes to multiple excel worksheets

python

pandas

xlsxwriter