从函数返回多个pandas.DataFrames并导出为不同的文件？

Question

def myfunc(arg1,arg2):
    df1 = pd.Datframe() #uses arg1 and arg2; df1 has 10cols & 100rows
    df2 = pd.Datframe() #uses arg1 and arg2; df2 has 4cols & 10rows
    return df1, df2

result = myfunc(arg1,arg2)

I need to extract the individual df from result and export as .xlsx

Answer 1

Your function will return a tuple, which is iterable. You can export the results like this:

import uuid
writer = pd.ExcelWriter('output.xlsx')

for r in result:
    r.to_excel(writer, str(uuid.uuid4()))

This will write your files in .xlsx format, and assign each file a unique name.

Hope this helps!

Answer 2

You can use sequence unpacking:

def myfunc(arg1,arg2):
    df1 = pd.Datframe() #uses arg1 and arg2; df1 has 10cols & 100rows
    df2 = pd.Datframe() #uses arg1 and arg2; df2 has 4cols & 10rows
    return df1, df2

df1, df2 = myfunc(arg1,arg2)

df1.to_excel('data1.xlsx')
df2.to_excel('data2.xlsx')

Or you can iterate in a loop:

for idx, df in enumerate(myfunc(arg1, arg2)):
    df.to_excel('data{0}.xlsx'.format(idx))