20200316 python 판다스(데이터 읽기_쓰기)

 

# read_csv(),to_csv() : csv 파일 텍스트 쓰기

import pandas as pd

#csv파일 저장
#index=False는 자동으로 추가된 인덱스 없이 저장
df = pd.DataFrame({'c0':[0,1,2],'c1':[1,2,3],'c2':[4,5,6],'c3':[7,8,9]})
df.to_csv("test.csv", index=False)
print(df)
print('\n')

#읽기
df1 = pd.read_csv("test.csv")
print(df1)
print('\n')

#index=False를 지정하지 않았을경우 0,1,2 열이 하나 더 생김
#header가 없을 경우 header=None
df2 = pd.read_csv("test.csv",header=None)
print(df2)
print('\n')

#index_col='c0'을 옵션으로 줄 경우 c0열을 인덱스로 불러옴
df3 = pd.read_csv("test.csv",index_col='c0')
print(df3)
print('\n')

#열 이름을 따로 설정
df4 = pd.read_csv("test.csv",names=['1열','2열','3열','4열'])
print(df4)

#read_excel() : 엑셀 파일 읽기
import pandas as pd
df1 = pd.read_excel("남북한발전전력량.xlsx")
df1.head()

# read_json() : 제이슨 파일 읽기
import pandas as pd

df = pd.read_json("json_test.json")
print(df)
print('\n')
print(df.index)
print(df.columns)

# ExcelWriter :  여러개의 데이터프레임을 하나의 엑셀파일로 저장하기
import pandas as pd
data1 = {'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],
        'c3':[7,8,9],
        'c4':[10,11,12],
        'c5':[13,14,15]}

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

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

print(df1)
print('\n')
print(df2)

#pd.ExcelWriter("test_excel.xlsx")쓸 준비가 되어있다.
writer = pd.ExcelWriter("test_excel.xlsx")
df1.to_excel(writer, sheet_name='1번 시트')
df2.to_excel(writer, sheet_name='2번 시트')
#이용을 마무리한다. 따로 저장해주지 않으면 덮어씌어진다.
writer.save()