파이썬, pandas를 이용하여 엑셀 파일 읽기, 추가, 저장하기

파이썬, pandas를 이용하여 엑셀 파일 읽기, 추가, 저장하기

글. 오상문 sualchi@daum.net

 

주의할 것은 엑셀 파일에서 컬럼명이 한글인 경우 에러가 발생할 수 있으니 영문을 사용하기 바랍니다.

 

import pandas as pd
df = pd.read_excel('data.xlsx')  # 엑셀 파일 읽기 
print(df['Company'])  # Company 컬럼 출력하기 
'''
----
0    가상사
1    나상사
2    다상사
Name: Company, dtype: object
----
'''
print(df['Count'])   # Count 컴럼 출력하기 
'''
----
0    1
1    2
2    3
Name: Count, dtype: int64
----
'''
print(df)   # 전체 보기 
'''
----
  Company  Count
0     가상사      1
1     나상사      2
2     다상사      3
----
'''
# rows = pd.DataFrame({'Company':['West','North','South'],'Count':[10,20,30]})  # 여러 줄 추가
row =[{'Company':'EastCo','Count':10}]   # 한 줄 추가 
df = df.append(row, 'sort=False')
writer = pd.ExcelWriter('save.xlsx', engine='openpyxl')  # 저장할 파일 정보 생성 
df.to_excel(writer, index=False)  
# df.to_excel(writer, startrow=len(df)+2, index=False)  # 삽입 위치 지정하여 추가하기 
writer.save()  # 엑셀 파일 저장 

 

save 엑셀 파일을 열어보면 다음처럼 줄이 추가된 것을 볼 수 있습니다.

 

  Company  Count
0     가상사      1
1     나상사      2
2     다상사      3

3     EastCo     10