[Python] 파이썬을 이용한 엑셀 자동화 프로그램 만들기

파이썬을 이용한 엑셀 파일 자동화에 대해서 알아보도록 하겠습니다. 엑셀 작업을 자주 하시는 분들이라면 반복적인 작업에 지치신 적이 있으실 겁니다. 파이썬을 이용하면 이런 지루한 작업들을 쉽고 빠르게 처리할 수 있습니다.

파이썬으로 엑셀 다루기

먼저, 파이썬에서 엑셀 파일을 다루기 위해서는 특별한 라이브러리가 필요해요. 가장 많이 사용되는 라이브러리는 'openpyxl'이에요. 이 라이브러리를 사용하면 엑셀 파일을 쉽게 읽고, 수정하고, 새로 만들 수 있답니다.

openpyxl 설치하기

openpyxl을 사용하기 위해서는 먼저 설치를 해야 합니다. 터미널이나 명령 프롬프트에서 다음 명령어를 입력해주세요.

pip install openpyxl

이렇게 하면 openpyxl이 설치가 됩니다.

엑셀 파일 읽기

자, 이제 openpyx을 이용해 엑셀 파일을 읽어볼까요? 다음과 같은 코드를 사용하시면 됩니다.

엑셀파일 경로는 파이썬 파일 경로와 동일한 곳에 저장하면 됩니다.

example.xlsx 캡처본(좌), 출력결과(우)

from openpyxl import load_workbook

# 엑셀 파일 열기
workbook = load_workbook('example.xlsx')

# 활성화된 시트 선택
sheet = workbook.active

# 특정 셀의 값 읽기
print(sheet['A1'].value)

# 여러 셀 읽기
for row in sheet['A1:C5']:
    for cell in row:
        print(cell.value)

이 코드는 'example.xlsx' 파일을 열고, 활성화된 시트에서 데이터를 읽어옵니다.

데이터 수정하고 새 파일 만들기

엑셀 파일을 읽는 것뿐만 아니라, 데이터를 수정하고 새로운 파일을 만들 수도 있어요. 아래 코드를 한번 볼까요?

from openpyxl import Workbook

# 새 워크북 생성
workbook = Workbook()
sheet = workbook.active

# 데이터 추가
sheet['A1'] = '이름'
sheet['B1'] = '나이'
sheet['A2'] = '홍길동'
sheet['B2'] = 30

# 파일 저장
workbook.save('new_file.xlsx')

이 코드는 새로운 엑셀 파일을 만들고, 거기에 데이터를 추가한 뒤 저장합니다.

대량의 데이터 처리하기

엑셀 자동화의 진정한 힘은 대량의 데이터를 처리할 때 발휘 됩니다. 예를 들어, 수천 개의 행을 가진 엑셀 파일에서 특정 조건을 만족하는 데이터만 추출하거나, 여러 파일의 데이터를 하나로 합치는 작업을 순식간에 할 수 있어요.

import openpyxl

# 여러 엑셀 파일에서 데이터 추출
def extract_data(files):
    all_data = []
    for file in files:
        wb = openpyxl.load_workbook(file)
        sheet = wb.active
        for row in sheet.iter_rows(min_row=2, values_only=True):
            if row[2] > 50000:  # 예: 급여가 50000 이상인 직원만 추출
                all_data.append(row)
    return all_data

# 새 파일에 데이터 저장
def save_data(data, filename):
    wb = openpyxl.Workbook()
    sheet = wb.active
    for row in data:
        sheet.append(row)
    wb.save(filename)

# 실행
files = ['employee1.xlsx', 'employee2.xlsx', 'employee3.xlsx']
extracted_data = extract_data(files)
save_data(extracted_data, 'high_salary_employees.xlsx')

이 코드는 여러 엑셀 파일에서 조건에 맞는 데이터를 추출하여 새로운 파일로 저장 합니다.

차트와 그래프 만들기

데이터 분석에서 빼놓을 수 없는 것이 바로 시각화 입니다. openpyxl을 사용하면 엑셀에서 차트와 그래프도 만들수 있어요.

실행결과

from openpyxl import Workbook
from openpyxl.chart import BarChart, Reference

wb = Workbook()
sheet = wb.active

# 데이터 추가
data = [
    ['과일', '판매량'],
    ['사과', 50],
    ['바나나', 30],
    ['오렌지', 40]
]

for row in data:
    sheet.append(row)

# 차트 생성
chart = BarChart()
chart.title = "과일 판매량"
chart.y_axis.title = '판매량'
chart.x_axis.title = '과일'

data = Reference(sheet, min_col=2, min_row=1, max_row=4, max_col=2)
cats = Reference(sheet, min_col=1, min_row=2, max_row=4)
chart.add_data(data, titles_from_data=True)
chart.set_categories(cats)

sheet.add_chart(chart, "E1")

wb.save("fruit_sales_chart.xlsx")

이 코드는 간단한 막대 그래프를 만들어 엑셀 파일에 추가합니다.

마무리

지금까지 파이썬을 이용한 엑셀 파일 자동화에 대해서 알아보았습니다. 이제 여러분도 파이썬을 활용해 지루하고 반복적인 엑셀 작업을 자동화 할 수 있을거에요. 처음에는 어려워 보일 수 있지만, 조금만 연습하면 금방 익숙해질 거예요.

 

파이썬의 강력한 기능을 활용하면  데이터 정리, 분석, 시각화 등 다양한 작업을 효율적으로 처리할 수 있답니다. 여러분의 업무 효율성이 크게 향상될 거에요.

 

마지막으로 파이썬과 엑셀을 함께 사용하면서 주의해야할 사항이 있습니다. 큰 파일을 다룰 때는 메모리 사용에 주의해야 하고, 중요한 데이터를 수정할 때는 항상 백업을 먼저 해두는 것이 좋답니다.