Python XlsxWriter 모듈로 엑셀 시트에 파이 차트 추가하는 방법

파이썬은 기본 라이브러리 외에도 개인 개발자들이 만든 다양한 외부 라이브러리를 통해 기능을 확장할 수 있습니다. 그중 XlsxWriter는 대표적인 라이브러리로, 파이썬 프로그램의 데이터를 엑셀 파일로 생성할 수 있을 뿐만 아니라 차트까지 손쉽게 만들어 줍니다.

파이 차트(Pie Chart) 만들기

아래 예제에서는 XlsxWriter를 활용해 파이 차트를 생성하는 방법을 살펴보겠습니다.

작업 순서는 다음과 같습니다.

1. 먼저 워크북(workbook) 객체를 생성합니다.
2. 워크북에 워크시트(worksheet)를 추가합니다.
3. 차트에 사용할 데이터를 정의하고, 데이터가 저장될 엑셀 열 위치를 지정합니다.
4. 해당 데이터 범위를 기반으로 파이 차트를 정의합니다.
5. 마지막으로 완성된 차트를 워크시트의 원하는 위치에 삽입합니다.

예제 코드

import xlsxwriter

workbook = xlsxwriter.Workbook('pie_chart_example.xlsx')
worksheet = workbook.add_worksheet()

# 차트에 표시할 데이터 추가
data = [
    ['milk', 'fruit', 'eggs', 'grains'],
    [27, 34, 12, 8]
]
worksheet.write_column('A1', data[0])
worksheet.write_column('B1', data[1])

# 새 차트 객체 생성
chart = workbook.add_chart({'type': 'pie'})

# 차트에 시리즈(데이터 계열) 추가
chart.add_series({
    'categories': '=Sheet1!$A$1:$A$4',
    'values': '=Sheet1!$B$1:$B$4'
})

# 워크시트의 특정 위치에 차트 삽입
worksheet.insert_chart('C5', chart)

workbook.close()

코드 설명

• write_column: A열에는 항목 이름(milk, fruit, eggs, grains)을, B열에는 각 항목의 값(27, 34, 12, 8)을 작성합니다.
• add_chart: {'type': 'pie'} 옵션으로 파이 차트 유형을 지정합니다.
• add_series: categories는 차트의 조각 이름이 될 범위, values는 각 조각의 크기를 결정하는 값 범위입니다.
• insert_chart: 차트를 C5 셀 위치부터 시작하도록 배치합니다.

위 코드를 실행하면 아래와 같은 결과 화면을 확인할 수 있습니다.