본문 바로가기
Study/혼자 공부하는 판다스

혼자 공부하는 판다스 - 데이터 저장하기

by Wanooky 2022. 4. 4.

데이터를 불러왔으니 데이터를 저장도 해보자.

 

처음은 csv 파일로 저장하는 것이다.

 

DataFrame.to_csv(파일이름, 경로)

 

import pandas as pd
data = {'name': ['Jerry', 'Riah', 'Paul'],
		'algol':['A', 'A+', 'B'],
        'basic' : ['C','B','B+'],
        'c++' : ['B+','C','C+']
        }

df = pd.DataFrame(data)
df.set_index('name',inplace=True)
print(df)

df.to_csv('./df_sample.csv')

두 번째는 json 파일로 저장하는 것이다

 

DataFrame.to_json(파일이름(경로))

 

data = {'name' : ['Jerry','Riah', 'Paul'],
        'algol':['A','A+','B'],
        'basic' : ['C','B','B+'],
        'c++' : ['B+','C','C+']
        }

df = pd.DataFrame(data)
df.set_index('name',inplace=True)
print(df)

df.to_json('./df_sample.json')

 

세 번째는 excel 파일로 저장하는 것이다.

 

DataFrame.to_excel(파일이름(경로))

data = {'name' : ['Jerry','Riah', 'Paul'],
        'algol':['A','A+','B'],
        'basic' : ['C','B','B+'],
        'c++' : ['B+','C','C+']
        }

df = pd.DataFrame(data)
df.set_index('name',inplace=True)
print(df)

df.to_excel('./df_sample.xlsx')

 

여러 개의 데이터 프레임을 하나의 Excel 파일로 저장하는 방법이 있다

 

pandas.ExcelWriter('파일 이름(경로)')

 

ata1 = {'name' : ['Jerry','Riah', 'Paul'],
        'algol':['A','A+','B'],
        'basic' : ['C','B','B+'],
        'c++' : ['B+','C','C+']
        }

data2 = {'c0':[1,2,3],
         'c1':[4,5,6],
         'c2':[7,8,9],
         'c3':[10,11,12],
         'c4':[13,14,15]}

df1 = pd.DataFrame(data1)
df1.set_index('name',inplace=True)
print(df1)

df2 = pd.DataFrame(data2)
df2.set_index('c0',inplace=True)
print(df2)

writer = pd.ExcelWriter('./df_excelwriter.xlsx')
df1.to_excel(writer, sheet_name='sheet1')
df2.to_excel(writer, sheet_name='sheet2')
writer.save()