Matplotlib 을 사용한 그룹 막대 차트 생성

Beginner

This tutorial is from open-source community. Access the source code

소개

이 랩에서는 Matplotlib 을 사용하여 단위가 있는 그룹 막대 차트를 만드는 방법을 배우게 됩니다. 막대 차트는 그룹 및 음료 선택에 따른 컵 높이를 표시합니다. 이 튜토리얼은 Python 코드를 사용하여 차트를 만드는 단계를 안내합니다.

VM 팁

VM 시작이 완료되면, 왼쪽 상단을 클릭하여 Notebook 탭으로 전환하여 실습을 위해 Jupyter Notebook에 접근하십시오.

때로는 Jupyter Notebook 이 로딩을 완료하는 데 몇 초 정도 기다려야 할 수 있습니다. Jupyter Notebook 의 제한으로 인해 작업의 유효성 검사는 자동화될 수 없습니다.

학습 중에 문제가 발생하면 Labby 에게 문의하십시오. 세션 후 피드백을 제공해주시면 문제를 신속하게 해결해 드리겠습니다.

필요한 라이브러리 가져오기

첫 번째 단계는 필요한 라이브러리를 가져오는 것입니다. 막대 차트를 만들기 위해 NumPy 와 Matplotlib 을 사용할 것입니다. cminch 단위는 단위를 변환하는 데 사용되는 basic_units.py 파일에 정의되어 있습니다.

import matplotlib.pyplot as plt
import numpy as np
from basic_units import cm, inch

데이터 정의

다음 단계는 막대 차트를 만드는 데 사용할 데이터를 정의하는 것입니다. 차를 위한 데이터 세트와 커피를 위한 데이터 세트, 두 개의 데이터 세트를 만들 것입니다. 각 세트는 각 그룹당 하나의 값, 총 다섯 개의 값을 갖습니다. 컵의 높이는 센티미터 (cm) 로 표시됩니다.

N = 5
tea_means = [15*cm, 10*cm, 8*cm, 12*cm, 5*cm]
tea_std = [2*cm, 1*cm, 1*cm, 4*cm, 2*cm]
coffee_means = (14*cm, 19*cm, 7*cm, 5*cm, 10*cm)
coffee_std = (3*cm, 5*cm, 2*cm, 1*cm, 2*cm)

Figure 및 Axes 객체 생성

다음 단계는 figure 및 axes 객체를 생성하는 것입니다. subplots() 함수를 사용하여 figure 및 axes 객체를 생성합니다.

fig, ax = plt.subplots()
ax.yaxis.set_units(inch)

막대 차트 매개변수 정의

다음 단계는 막대 차트의 매개변수를 정의하는 것입니다. 그룹의 x 위치, 막대의 너비, x-눈금 레이블을 정의합니다.

ind = np.arange(N)    ## the x locations for the groups
width = 0.35         ## the width of the bars
ax.set_xticks(ind + width / 2, labels=['G1', 'G2', 'G3', 'G4', 'G5'])

막대 차트 생성

다음 단계는 막대 차트를 생성하는 것입니다. bar() 함수를 사용하여 차트를 생성합니다. 차트를 위해 두 세트의 막대, 즉 차와 커피에 대한 막대를 생성합니다. 또한 차트에 오차 막대 (error bars) 를 추가합니다.

ax.bar(ind, tea_means, width, bottom=0*cm, yerr=tea_std, label='Tea')
ax.bar(ind + width, coffee_means, width, bottom=0*cm, yerr=coffee_std,
       label='Coffee')

차트에 레이블 및 제목 추가

마지막 단계는 차트에 레이블과 제목을 추가하는 것입니다. 차트 제목, x 축 레이블 및 차트 범례 (legend) 를 추가합니다.

ax.set_title('Cup height by group and beverage choice')
ax.set_xlabel('Group')
ax.legend()
ax.autoscale_view()

요약

이 랩 (lab) 에서는 Matplotlib 을 사용하여 단위 (unit) 가 있는 그룹 막대 차트를 만드는 방법을 배웠습니다. 필요한 라이브러리를 가져오고, 데이터를 정의하고, figure 및 axes 객체를 생성하고, 막대 차트 매개변수를 정의하고, 막대 차트를 생성하고, 차트에 레이블과 제목을 추가하는 방법을 배웠습니다. 이 지식을 통해 Matplotlib 을 사용하여 자신만의 막대 차트를 만들 수 있습니다.