Plotly - 막대그래프

 

 

Plotly¶

파이썬을 이용해 데이터 분석을 하면서 matplotlib, seaborn으로 그래프를 가장 많이 그렸다. 이때는 이 두 라이브러리 이외에는 그래프를 그릴 수 있는 라이브러리가 있다고는 생각을 못 했던 것 같다. 그러다 2021년 초에 인턴이 끝나고 집에서 놀고 있을 때 우연히 Plotly, Bokeh 같은 라이브러리들을 알게 되었고, 그래프가 예쁘게 그려지니깐 확실히 더 그래프를 그리는 맛이 생겼다.

Plotly로 그릴 수 있는 그래프에 대한 정보는 https://plotly.com/python/ 이곳에서 더 찾아볼 수 있으며, 지금은 간단하게 그래프를 그려보고자 기록을 남긴다.

 

Plotly 설치하기¶

In [ ]:

# plotly 설치하기
pip install plotly

In [ ]:

# 설치하면서 버전을 지정할 수도 있다.
pip install plotly==5.2.1

In [ ]:

# pip 명령어를 이용해 설치된 plotly 버전 업을 할 수도 있다.
pip install plotly --upgrade

 

만약 conda prompt에서 설치를 진행한다면 conda install plotly 명령어를 사용하면 된다.

In [6]:

import plotly

print("Plotly 버전 :", plotly.__version__)

 

Plotly 버전 : 5.1.0

 

먼저 그래프를 그리기 위해 사용한 데이터는 통계청(https://kosis.kr/index/index.do)에서 인구 총 조사 데이터를 이용했다.

이 데이터에서 2020년 총인구, 남자인구, 여자인구를 시도별로 가져왔다.

 

In [1]:

# 필요한 라이브러리 
import pandas as pd
import plotly.express as px

In [2]:

# 2020년 인구데이터 가져오기
data = pd.read_excel("2020년 인구.xlsx")

In [3]:

data.head(5)

Out[3]:

	행정구역별(읍면동)	2020	2020.1	2020.2
0	행정구역별(읍면동)	총인구 (명)	남자 (명)	여자 (명)
1	전국	51829136	25915207	25913929
2	서울특별시	9586195	4647733	4938462
3	부산광역시	3349016	1638751	1710265
4	대구광역시	2410700	1190271	1220429

 

엑셀 데이터를 그냥 가져오면 0번째 행에도 컬럼의 이름이 들어가기 때문에 옵션으로 header = 1을 추가한다.

아니면 엑셀로 열어서 위쪽 부분을 수정해도 된다.

In [4]:

# 2020년 인구데이터 가져오기
data = pd.read_excel("2020년 인구.xlsx", header = 1)

In [5]:

data.head(5)

Out[5]:

	행정구역별(읍면동)	총인구 (명)	남자 (명)	여자 (명)
0	전국	51829136	25915207	25913929
1	서울특별시	9586195	4647733	4938462
2	부산광역시	3349016	1638751	1710265
3	대구광역시	2410700	1190271	1220429
4	인천광역시	2945454	1477064	1468390

 

막대 그래프¶

먼저 간단하게 각 지역별 총 인구 데이터를 막대그래프로 그려본다.

In [6]:

fig = px.bar(data,                      # 그래프에 쓰일 데이터
             x='행정구역별(읍면동)',    # x축
             y='총인구 (명)',           # y축
             title = "2020년 총인구")   # 제목 

fig.show()

 

 

 

 

px.bar로 기본적인 막대 그래프를 그릴 수 있지만 막대의 색을 바꾼다거나, 누적막대그래프를 그리거나 할 때 추가적인 옵션들을 쓸 때는 graph_objects를 이용하는 것 같아 graph_objects를 주로 사용한다.

In [7]:

import plotly.graph_objects as go

fig = go.Figure(data=[
    go.Bar(name = '2020년 여자 인구', x = data["행정구역별(읍면동)"], y = data["여자 (명)"]), # 각각 그래프 이름, x축, y축
    go.Bar(name = '2020년 남자 인구', x = data["행정구역별(읍면동)"], y = data["남자 (명)"])
])

# 그래프의 모습을 group를 지정하면 두 개를 연달아 그려준다.
fig.update_layout(barmode = 'group')   
fig.show()

 

 

 

go.Bar()에 marker_color 옵션을 이용하면 각 막대의 색상을 바꾸는 것도 가능하다. 나는 http://www.w3big.com/ko/tags/html-colorname.html 이곳에서 색상의 이름과 색을 보고 맘에 드는 색상을 골라서 사용했다.

In [8]:

fig = go.Figure(data=[
    go.Bar(name = '2020년 여자 인구', x = data["행정구역별(읍면동)"], y = data["여자 (명)"], marker_color = 'PowderBlue'),
    go.Bar(name = '2020년 남자 인구', x = data["행정구역별(읍면동)"], y = data["남자 (명)"], marker_color = 'LightCoral')
])

fig.update_layout(barmode = 'group')   # 그래프의 모습을 group를 지정하면 두 개를 연달아 그려준다.
fig.show()

 

 

 

지금같은 경우는 x축의 눈금의 텍스트가 길어서 상관 없지만, 45도 정도로 기울인 x축 값들이 싫은 경우가 있을 수 있다. 이럴 때는 update_layout()에서 xaxis_tickangle 옵션에 각도를 설정해주면 된다.

In [9]:

fig = go.Figure(data=[
    go.Bar(name = '2020년 여자 인구', x = data["행정구역별(읍면동)"], y = data["여자 (명)"]), # 각각 그래프 이름, x축, y축
    go.Bar(name = '2020년 남자 인구', x = data["행정구역별(읍면동)"], y = data["남자 (명)"])
])

fig.update_layout(barmode = 'group',    # 그래프의 모습을 group를 지정하면 두 개를 연달아 그려준다.
                  xaxis_tickangle=-45)  # x축의 눈금의 각도를 조정해서 원하는 모습으로 출력할 수도 있다. 
fig.show()

 

 

 

위에서 두 막대 그래프를 연달아서 그렸다면 당연히 겹쳐서 그려보고 싶기도 하다. 이럴 때는 barmode 옵션을 group이 아닌 stack으로 바꿔주기만 하면 된다.

In [10]:

fig = go.Figure(data=[
    go.Bar(name = '2020년 여자 인구', x = data["행정구역별(읍면동)"], y = data["여자 (명)"]), # 각각 그래프 이름, x축, y축
    go.Bar(name = '2020년 남자 인구', x = data["행정구역별(읍면동)"], y = data["남자 (명)"])
])

fig.update_layout(barmode = 'stack')      # 그래프의 모습을 stack을 지정하면 두 개를 누적해서 그려준다.
fig.show()

 

 

 

만약 막대 그래프 위에 마우스를 올려서 해당 값을 확인하는게 귀찮다면, 막대그래프 위에 해당 값을 텍스트로 표현할 수도 있다.

In [11]:

fig = go.Figure(data=[
    go.Bar(name = '2020년 총 인구', 
           x = data["행정구역별(읍면동)"], 
           y = data["총인구 (명)"], 
           marker_color = 'LightSteelBlue', 
           text = data["총인구 (명)"])      # 텍스트로 표현할 값을 지정 - 여기선 y축과 동일 
])

fig.show()

 

 

 

출력되는 텍스트가 엉망진창이다. 어떤 거는 세로로 출력되고, 가로로 출력되고, 텍스트 크기도 제멋대로.....

이럴 때는 레이아웃의 파라미터를 설정해 주어야 한다.

In [12]:

fig = go.Figure(data=[
    go.Bar(name = '2020년 총 인구', 
           x = data["행정구역별(읍면동)"], 
           y = data["총인구 (명)"], 
           marker_color = 'LightSteelBlue', 
           text = data["총인구 (명)"])      # 텍스트로 표현할 값을 지정 - 여기선 y축과 동일 
])

fig.update_traces(texttemplate='%{text:.2s}',  # 표시될 텍스트의 형태 지정
                  textposition='outside')      # 막대그래프 밖에 텍스트 출력

# 텍스트 크기를 모두 같게 출력하고 싶으면 이 부분을 추가해야 한다. 
fig.update_layout(uniformtext_minsize=8,       # 최소 글꼴 크기 설정
                  uniformtext_mode='hide')     # 글꼴 크기에 맞지 않는 레이블에 대해 수행할 작업을 설정->show(오버플로우 표시),hide(숨김)

fig.show()

 

 

 

이번에는 간략하게 막대 그래프 기능을 다뤄볼려고 한다. 2020년 리그 오브 레전드 LCK Spring PoG 점수와 2021년 리그 오브 레전드 LCK Summer PoG 점수를 한 번 사용해본다.

In [13]:

# 2021년 LCK Spring Player of the Game(PoG)
players = ["ShowMaker", "Pyosik", "Chovy", "Keria", "Khan", "Canyon"]
points_21 = [1000, 1000, 1000, 900, 800, 700]

# 2020년 LCK Summer Player of the Game(PoG)
points_20 = [1100, 300, 1000, 800, 0, 1000]

In [14]:

colors = ['lightslategray',] * 6   # 색상을 담은 리스트를 생성한다.
colors[0] = 'crimson'              # 리스트의 첫 번째 색상만 cromsom으로 변경한다.

fig = go.Figure(data=[go.Bar(
    x = players,
    y = points_21,
    marker_color = colors       # 리스트를 색상으로 지정한다.
)])

fig.update_layout(title_text='2021 LCK Spring PoG')

 

 

In [15]:

colors = ['lightslategray',] * 6
colors[0] = 'crimson'

fig = go.Figure(data=[go.Bar(
    x = players,
    y = points_20,
    marker_color = colors
)])

fig.update_layout(title_text='2020 LCK Summer PoG')

 

 

 

막대그래프를 그리다보면 주식 차트처럼 등락률? 같이 표현하고 싶을 때도 있다. 그럴 때는 base 옵션을 사용하면 된다.

In [16]:

fig = go.Figure()
fig.add_trace(go.Bar(x = players, 
                     y = points_20,
                     base = [-1100, -300, -1000, -800, 0, -1000],   # 막대그래프의 기준을 설정해줄 수 있다. -> 0에 걸치도록
                     marker_color='crimson',
                     name='2020 Spring'))
fig.add_trace(go.Bar(x = players, 
                     y = points_21,
                     base=0,
                     marker_color='lightslategrey',
                     name='2021 Summer'))

fig.show()

 

 

 

각각의 막대그래프 넓이를 설정하는 것 또한 가능하다.

In [17]:

fig = go.Figure(data=[go.Bar(
    x = players,
    y = points_21,
    width = [1, 1, 1, 0.7, 0.5, 0.3] # 각 막대그래프의 넓이를 리스트에 넣어서 주면 된다.
)])

fig.show()

 

 

In [ ]: