pyecharts로 막대 그래프 그리기

데이터 시각화

pyecharts로 막대 그래프 그리기

크리넥스을라이스 2023. 1. 26. 20:24

안녕하세요 크리넥스을라이스입니다.

보통 파이썬을 이용해 데이터 시각화를 수행할 때는 matplotlib나 seaborn 라이브러리를 이용하실 겁니다.

많이 사용하는 라이브러리다 보니 저는 다소 질릴 때도 있었는데요

그래서 이번에 다뤄 볼 내용은 pyecharts 라이브러리로 어떻게 막대 그래프를 만들 수 있는지

한 번 정리해 보도록 하겠습니다.

(더 다양한 Pyecharts 라이브러리 그래프들은 이곳에서 살펴보실 수 있습니다.)

먼저 pyecharts를 설치해 주시면 됩니다.

pip install pyecharts

# 주피터노트북 셀
!pip install pyecharts

설치가 완료되었다면, 라이브러리를 import 해주시면 됩니다.

# 라이브러리 
import pyecharts

# 버전 확인
print("Pyecharts 버전 :", pyecharts.__version__)

pyecharts를 사용할 준비가 되었다면 이어서 시각화할 데이터를 찾아보겠습니다.

저는 2022년 KBO 팀 순위 정보를 사용해보도록 하겠습니다(링크).

마지막 홈과 방문 컬럼은 제외하고 데이터를 csv 파일로 만들어 사용했습니다.
(인코딩이 필요해 cp949 옵션을 추가해줬습니다.)

# 판다스 라이브러리 
import pandas as pd
# 그리고 싶은 그래프를 불러온다.
from pyecharts.charts import Bar
# 그래프 옵션을 설정하기 위해 필요한 라이브러리
from pyecharts import options as opts

# kbo csv 파일 읽어오기
kbo = pd.read_csv("./kbo.csv", encoding = "cp949")

# 데이터 출력
kbo

kbo 데이터를 x축은 팀, y축은 승으로 된 막대그래프로 그려보겠습니다.

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])

bar = (
    Bar()
    .add_xaxis(x)            # x축
    .add_yaxis("승리", y)    # y축
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위",  # 제목
                                                 subtitle = "정규시즌"))       # 소제목
    .render_notebook()       # 주피터 노트북에 그래프를 바로 출력
)

bar

실행시키시면 아래 이미지처럼 막대그래프가 출력됩니다.

막대 위에 마우스를 가져가면 해당 막대의 정보를 확인하실 수 있습니다.

만약 그래프를 html 파일로 저장하고 싶다면 다음과 같이 .render_notebook()을

render("경로/파일명.html")으로 바꿔주시면 됩니다.

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])

bar = (
    Bar()
    .add_xaxis(x)            # x축
    .add_yaxis("승리", y)    # y축
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위",  # 제목
                                                 subtitle = "정규시즌"))       # 소제목
    .render("kbo_bar_graph.html")     # 그래프를 html로 저장
)

막대 그래프 위의 숫자를 출력하고 싶지 않다면 add_yaxis에서 label_opts 옵션을 False로 주면 됩니다.

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])

bar = (
    Bar()
    .add_xaxis(x)            # x축
    .add_yaxis("승리", y, label_opts = False)  # label_opts = False 옵션으로 데이터 출력을 off
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위",  # 제목
                                                 subtitle = "정규시즌"))       # 소제목
    .render_notebook()       # 주피터 노트북에 그래프를 바로 출력
)

bar

빨간색이 너무 부담스러워 다른 색으로 바꾸고 싶다면 color 옵션을 설정해 주면 됩니다.

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])

bar = (
    Bar()
    .add_xaxis(x)            # x축
    .add_yaxis("승리", y, label_opts = False, color = "CornflowerBlue")  # color 옵션으로 색상을 변경
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위",  # 제목
                                                 subtitle = "정규시즌"))       # 소제목
    .render_notebook()       # 주피터 노트북에 그래프를 바로 출력
)

bar

만약 x축과 y축에 대한 이름을 넣고 싶다면 다음과 같이 옵션을 추가해 주시면 됩니다.

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])

bar = (
    Bar()
    .add_xaxis(x)            # x축
    .add_yaxis("승리", y, label_opts = False, color = "LightCoral")  # color 옵션으로 색상을 변경
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위", subtitle = "정규시즌"),
                                        xaxis_opts = opts.AxisOpts(name = "참여 팀명"),  # X축 이름
                                        yaxis_opts = opts.AxisOpts(name = "승리 횟수"))  # Y축 이름
    .render_notebook()       # 주피터 노트북에 그래프를 바로 출력
)

bar

막대 그래프에 대한 정보를 더 넣고 싶다면 y축에 눈금에 대한 정보를 추가할 수도 있습니다.

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])

bar = (
    Bar()
    .add_xaxis(x)            # x축
    .add_yaxis("승리", y, label_opts = False, color = "LightCoral")  # color 옵션으로 색상을 변경
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위", subtitle = "정규시즌"),
                                        xaxis_opts = opts.AxisOpts(name = "참여 팀명"),  # X축 이름
                                        yaxis_opts = opts.AxisOpts(name = "승리 횟수",
                                                                   axislabel_opts = opts.LabelOpts(formatter="{value} (경기)"))) # Y축 눈금 포맷
    .render_notebook()       # 주피터 노트북에 그래프를 바로 출력
)

bar

결과를 보시면 왼쪽 y축에 '경기'라는 문구가 추가된 것을 볼 수 있습니다.

다음은 그래프에 줌/아웃 기능을 추가하는 옵션입니다.

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])

bar = (
    Bar()
    .add_xaxis(x)            # x축
    .add_yaxis("승리", y, label_opts = False, color = "LightCoral")  # color 옵션으로 색상을 변경
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위", subtitle = "정규시즌"),
                                        xaxis_opts = opts.AxisOpts(name = "참여 팀명"),  # X축 이름
                                        yaxis_opts = opts.AxisOpts(name = "승리 횟수",
                                                                   axislabel_opts = opts.LabelOpts(formatter="{value} (경기)")),
                                        datazoom_opts=[opts.DataZoomOpts(), opts.DataZoomOpts(type_="inside")])  # 그래프를 줌/아웃 할 수 있다
    .render_notebook()       # 주피터 노트북에 그래프를 바로 출력
)

bar

그래프 오른쪽 상단에 필요한 기능을 담은 툴박스를 만드는 옵션입니다.

(다만, 툴 기능들이 중국어로 출력되고 있네요.)

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])

bar = (
    Bar()
    .add_xaxis(x)            # x축
    .add_yaxis("승리", y, label_opts = False, color = "LightCoral")  # color 옵션으로 색상을 변경
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위", subtitle = "정규시즌"),
                                        xaxis_opts = opts.AxisOpts(name = "참여 팀명"),  # X축 이름
                                        yaxis_opts = opts.AxisOpts(name = "승리 횟수",
                                                                   axislabel_opts = opts.LabelOpts(formatter="{value} (경기)")),
                                        datazoom_opts=[opts.DataZoomOpts(), opts.DataZoomOpts(type_="inside")],
                                        toolbox_opts = opts.ToolboxOpts())          # 그래프 오른쪽 위에 툴박스를 생성한다.
    .render_notebook()       # 주피터 노트북에 그래프를 바로 출력
)

bar

이번에는 승리와 패배를 같이 막대 그래프로 표현해보겠습니다.

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])
z = list(kbo["패"])  # 패 컬럼을 리스트로 생성

double_bar = (
    Bar()
    .add_xaxis(x)        
    .add_yaxis("승리", y)
    .add_yaxis("패배", z)    # z를 y축에 추가
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위",  # 제목
                                                 subtitle = "정규시즌"))       # 소제목
    .render_notebook()
)

double_bar

상단 중앙의 승리/패배를 눌러주시면 해당 항목은 그래프에서 사라지는 것을 확인하실 수 있습니다.

막대 사이의 간격을 조정하고 싶다면 gap 옵션을 사용하시면 됩니다.

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])
z = list(kbo["패"])

double_bar = (
    Bar()
    .add_xaxis(x)        
    .add_yaxis("승리", y, gap="0%")  # 막대 간격 조정
    .add_yaxis("패배", z, gap="0%")  # 막대 간격 조정
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위",  # 제목
                                                 subtitle = "정규시즌"))       # 소제목
    .render_notebook()
)

double_bar

누적 막대 그래프로 표현하고 싶다면 stack 옵션을 사용하면 됩니다.

# 리스트로 변경
x = list(kbo["팀명"])
y = list(kbo["승"])
z = list(kbo["패"])
a = list(kbo["무"])

double_bar = (
    Bar()
    .add_xaxis(x)        
    .add_yaxis("승리", y, label_opts = False, color = "LightBlue", stack="stack") # stack 옵션
    .add_yaxis("패배", z, label_opts = False, color = "LemonChiffon", stack="stack") # stack 옵션
    .add_yaxis("무", a, label_opts = False, color = "LightCoral", stack="stack") # stack 옵션
    .set_global_opts(title_opts = opts.TitleOpts(title = "2022년 KBO 팀순위",
                                                 subtitle = "정규시즌"))   
    .render_notebook()
)

double_bar

오늘은 여기까지 준비했습니다.

감사합니다.