[python+pandas] pandas로 엑셀 파일 읽고 수정한 후 쓰기

오늘은 pandas를 이용해서 엑셀 파일을 다루는 법에 대해 살펴보도록 하겠습니다.

 

다음과 같은 내용을 Sheet1에 담고 있는 test.xlsx 파일이 있습니다. 

 

 

우선 이것을 읽어서 pandas의 dataframe에 넣겠습니다. 여기까지 필요한 코드는 다음과 같습니다. 참고로 아래 코드를 실행하려면 pandas, xlrd, openpyxl 모듈이 필요합니다. 

 

pip install pandas

pip install xlrd

pip install openpyxl 

 

등의 방식으로 설치해주세요. 

 

import pandas as pd
 
df = pd.read_excel('test.xlsx', sheet_name="Sheet1", engine='openpyxl')
 
print(df)

 

위 코드를 실행하면 다음과 같은 결과가 출력됩니다. 엑셀 파일의 Sheet1을 잘 읽어낸 것을 확인하실 수 있습니다. 

 

 

이제 저는 각각 주식 투자로 인해 전재산의 절반을 날렸다고 가정하고, 전재산을 반으로 줄인 후에 test.xlsx에 덮어쓰겠습니다. 

 

import pandas as pd
 
df = pd.read_excel('test.xlsx', sheet_name="Sheet1", engine='openpyxl')
 
print(df)
 
df['전재산'] = df['전재산']/2
 
print(df)
 
df.to_excel('test.xlsx', sheet_name="Sheet1", index=False)

 

위 코드를 실행하면 다음과 같은 화면이 출력되고,

 

 

생성된 test.xlsx 안에는 다음과 같은 내용이 담겨 있습니다.