[파이썬] 엑셀 파일 다루기

파이썬에서 엑셀파일을 다루는 것은 데이터 분석, 자동화 스크립트 작성, 리포트 생성 등 다양한 작업에 있어 매우 유용합니다. 파이썬에는 'pandas', 'openpyxl', 'xlrd', 'xlwt' 등의 라이브러리를 통해 엑셀 파일을 읽고, 쓰고, 수정할 수 있습니다.

 

1.  pandas를 사용한 엑셀 파일 처리

pandas는 데이터 분석을 위한 고수준의 API를 제공하며, read_excel() 및 to_excel() 함수를 통해 간단히 엑셀 파일을 읽고 쓸 수 있습니다.

 

엑셀 파일 읽기

import pandas as pd

# 엑셀 파일 읽기
df = pd.read_excel('example.xlsx', sheet_name='Sheet1')

# 데이터 확인
print(df)

 

엑셀 파일 쓰기

# 데이터프레임 생성
data = {'Name': ['John', 'Anna'],
        'Age': [28, 22],
        'City': ['New York', 'Paris']}
df = pd.DataFrame(data)

# 엑셀 파일 쓰기
df.to_excel('output.xlsx', sheet_name='Sheet1', index=False)

 

2. openpyxl을 사용한 엑셀 파일 처리

openpyxl 라이브러리는 엑셀 파일의 더 세부적인 요소들을 다룰 수 있게 해주며, 엑셀 파일을 생성하고, 셀에 접근하고, 스타일을 적용하는 등의 작업이 가능합니다.

 

엑셀 파일 읽기

from openpyxl import load_workbook

# 엑셀 파일 로드
wb = load_workbook('example.xlsx')
sheet = wb['Sheet1']

# 셀 값 읽기
print(sheet['A1'].value)
print(sheet.cell(row=1, column=2).value)

# 모든 행 순회
for row in sheet.iter_rows(values_only=True):
    print(row)

 

엑셀 파일 쓰기

from openpyxl import Workbook

# 새 워크북 생성
wb = Workbook()
sheet = wb.active  # 활성 시트 가져오기

# 셀에 값 쓰기
sheet['A1'] = 'Name'
sheet['B1'] = 'Age'

# 행 추가
sheet.append(['John', 28])
sheet.append(['Anna', 22])

# 파일 저장
wb.save('new_example.xlsx')

 

사용 목적에 따라 pandas와 openpyxl을 선택해서 사용하면 된다.
- pandas: 데이터 분석
- openpyxl: 세부적인 스타일링이나 복잡한 작업