Python将数据框切片为许多excel文件

Question

我想将一个数据框拆分为多个 excel 文件，使用其行值的名称作为文件的名称

示例数据帧

    account_key    State
0    111            NY
1    111            CA
2    222            NJ
3    888            WE
4    888            TL

预期结果：

第一个excel文件名为account_111.xlsx，里面的数据是

account_key    State
 111            NY
 111            CA

第二个excel文件名为account_222.xlsx，里面的数据是

account_key    State
222             NJ

我的代码没有给我预期的结果

names = df['account_key'].unique().tolist()
writer = pd.ExcelWriter(path + 'my.xlsx', engine='xlsxwriter')
for myname in names:
    mydf = df.loc[df['account_key']==myname]
    mydf.to_excel(writer)
    writer.save()    

顺便说一句，如何修改代码以在循环时将工作表从“sheet1”重命名为“myNote”？

Answer 1

IIUC，只需在groupby 中使用to_excel

for acc, g in df.groupby('account_key'):
    g.to_excel(f'account_{acc}.xlsx')

如果您有不支持f-strings 的旧版python，请使用

for acc, g in df.groupby('account_key'):
    g.to_excel('account_{}.xlsx'.format(acc))

要命名工作表，只需将名称作为第二个参数传递

g.to_excel(f'account_{acc}.xlsx', 'myNote')