파이썬으로 엑셀 자동화하는 간단한 코드 예시

최근 들어 데이터 처리와 분석 업무가 점점 더 중요해지고 있습니다. 특히 엑셀은 많은 사람들이 사용하는 데이터 관리 도구 중 하나로, 많은 기업과 개인들이 엑셀을 통해 데이터를 정리하고 분석하는 데에 큰 의존을 하고 있습니다. 하지만 반복적인 작업을 엑셀에서 수동으로 수행하는 것은 시간과 노력이 매우 많이 소모됩니다. 이런 문제를 해결하기 위해 많은 사람들은 파이썬을 사용하여 엑셀 자동화를 시도하고 있습니다. 이번 글에서는 파이썬을 활용하여 엑셀 파일을 자동으로 생성하고 편집하는 방법에 대해 살펴보겠습니다.

파이썬 환경 설정하기

엑셀 파일을 다루기 위해서는 먼저 파이썬 환경을 준비해야 합니다. 기본적으로 ‘openpyxl’이라는 라이브러리를 설치해야 합니다. 이 라이브러리는 엑셀 파일을 읽고 쓰는 데 필요한 기능을 제공해 주며, 설치는 다음의 간단한 명령어로 가능합니다.

pip install openpyxl

이제 라이브러리가 준비되었으니, 실제로 엑셀 파일을 생성하고 데이터를 추가하는 코드를 작성해 보겠습니다.

엑셀 파일 생성 및 데이터 입력

엑셀 파일을 생성하고 데이터 입력을 위해 아래와 같은 코드를 사용하실 수 있습니다.

from openpyxl import Workbook
# 워크북 생성
wb = Workbook()
# 기본 시트 선택
ws = wb.active
ws.title = "데이터"
# 데이터 입력
ws['A1'] = "이름"
ws['B1'] = "나이"
ws['C1'] = "도시"
# 몇 가지 예시 데이터 추가
data = [
  ("홍길동", 30, "서울"),
  ("김영희", 25, "부산"),
  ("이철수", 40, "인천")
]
for row in data:
  ws.append(row)
# 파일 저장
wb.save("sample_data.xlsx")

위 코드에서 사용된 ‘Workbook’ 클래스를 통해 새로운 엑셀 파일을 생성하고, 활성화된 시트에 데이터를 추가하는 과정을 보여주고 있습니다. 예시로 ‘이름’, ‘나이’, ‘도시’라는 열 제목이 생성되고, 아래에 데이터가 추가됩니다.

엑셀 파일에서 데이터 읽기

이제 저장된 엑셀 파일에서 데이터를 읽어오는 방법을 살펴보겠습니다. 다음 코드를 사용하여 엑셀 파일에서 데이터를 읽어올 수 있습니다.

from openpyxl import load_workbook
# 저장된 엑셀 파일 열기
wb = load_workbook("sample_data.xlsx")
# 시트 선택
ws = wb['데이터']
# 데이터 읽기
for row in ws.iter_rows(min_row=2, values_only=True):
  print(row)

위 코드는 저장된 엑셀 파일을 열고, ‘데이터’ 시트를 선택한 후, 2행부터 시작하여 모든 데이터를 읽어 출력하는 과정을 보여줍니다. ‘iter_rows’ 메서드는 지정된 범위 내의 데이터를 행 단위로 반환합니다.

조건부 데이터 처리하기

많은 경우, 데이터를 읽어오고 나서 특정 조건에 따라 처리를 해야 하는 경우가 많습니다. 예를 들어, 나이가 30세 이상인 사람들만 필터링하고 싶다면 다음과 같은 코드를 사용할 수 있습니다.

for row in ws.iter_rows(min_row=2, values_only=True):
  name, age, city = row
  if age >= 30:
    print(f"{name}님은 {city}에 거주하며, 나이는 {age}세입니다.")

이 코드는 각 행에서 나이 정보를 확인하여 30세 이상인 사람에 대해 해당 정보를 출력하고 있습니다. 이러한 방식으로 특정 조건에 따라 데이터 처리를 수행할 수 있습니다.

엑셀 파일 수정하기

기존의 엑셀 파일에 데이터를 추가하거나 수정하는 것도 가능합니다. 다음 코드는 특정 셀의 값을 수정하는 예제입니다.

# 기존 파일 열기
wb = load_workbook("sample_data.xlsx")
ws = wb['데이터']
# 특정 셀 수정
ws['B2'] = 31 # 홍길동의 나이를 수정
# 변경된 내용을 저장
wb.save("sample_data.xlsx")

위의 예제에서는 ‘B2’ 셀에 해당하는 값을 수정하여 홍길동의 나이를 변경하고, 변경된 내용을 다시 파일에 저장합니다.

결론

파이썬을 활용하면 엑셀 파일의 생성, 읽기, 수정 등이 매우 간편하게 이루어질 수 있습니다. 이러한 자동화는 반복적인 작업을 줄여주고, 효율적인 데이터 관리를 가능하게 합니다. 위에서 소개한 예제들을 바탕으로 자신만의 데이터 처리 자동화 스크립트를 작성해 보시면 좋겠습니다. 끝으로 엑셀 자동화를 통해 많은 시간을 절약하고, 업무의 효율성을 높여보세요!

자주 묻는 질문 FAQ