pandas를 이용하여 두 개의 입력으로 두 개의 excel sheet 생성하기

아래에 간단한 프로그램 보기를 참고하시기 바람.

 

 

 

import pandas as pd

def fir():

    a={'a':[1,2], 'b':[3,4]}

    dfa=pd.DataFrame(a)

    return dfa

def sec():

    b={'c':[1,2], 'd':[3,4]}

    dfb=pd.DataFrame(b)

    return dfb def

 

result(argument):

    if argument == '1':

        df=fir()

        df.to_excel('D:/kang.xlsx')

    if argument == '2':

        df1=fir()

        df2=sec()

        with pd.ExcelWriter('D:/kang.xlsx') as writer:

            df1.to_excel(writer, sheet_name='Sheet_name_1')

            df2.to_excel(writer, sheet_name='Sheet_name_2')

 

result('2') # result 인수에 2가 입력되면 dfa와 dfb가 kang.xlsx excel 문서의 sheet_name_1과 sheet_name_2에 

입력된다.