Matplotlib 를 사용한 플롯 생성

Beginner

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

소개

이 랩에서는 Python Matplotlib 를 사용하여 플롯을 생성하는 과정을 안내합니다. Matplotlib 는 Python 프로그래밍 언어를 위한 플로팅 라이브러리입니다. 이 랩에서는 색상, 선 너비 등 플롯의 속성을 사용자 정의하는 방법을 배우게 됩니다.

VM 팁

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

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

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

필요한 모듈 임포트

먼저, 필요한 모듈을 임포트해야 합니다. 이 경우, matplotlib.pyplotnumpy를 임포트해야 합니다.

import matplotlib.pyplot as plt
import numpy as np

속성 사이클 정의 및 색상 가져오기

다음으로, 속성 사이클을 정의하고 거기에서 색상을 가져와야 합니다.

prop_cycle = plt.rcParams['axes.prop_cycle']
colors = prop_cycle.by_key()['color']

선 너비 정의

이제, 플롯에 대한 선 너비를 정의합니다.

lwbase = plt.rcParams['lines.linewidth']
thin = lwbase / 2
thick = lwbase * 3

서브플롯 생성

2x2 그리드의 서브플롯을 생성합니다.

fig, axs = plt.subplots(nrows=2, ncols=2, sharex=True, sharey=True)

수평 및 수직선 추가

이제, 속성 순환 (property cycle) 의 색상을 사용하여 각 서브플롯에 수평 및 수직선을 추가합니다.

for icol in range(2):
    if icol == 0:
        lwx, lwy = thin, lwbase
    else:
        lwx, lwy = lwbase, thick
    for irow in range(2):
        for i, color in enumerate(colors):
            axs[irow, icol].axhline(i, color=color, lw=lwx)
            axs[irow, icol].axvline(i, color=color, lw=lwy)

서브플롯 사용자 정의

아래쪽 서브플롯의 배경색을 검정색으로 설정하고, x 축 눈금을 설정하며, 각 서브플롯에 제목을 추가하여 서브플롯을 사용자 정의합니다.

axs[1, icol].set_facecolor('k')
axs[1, icol].xaxis.set_ticks(np.arange(0, 10, 2))
axs[0, icol].set_title(f'line widths (pts): {lwx:g}, {lwy:g}',
                       fontsize='medium')

y 축 눈금 사용자 정의

가장 왼쪽 서브플롯의 y 축 눈금을 사용자 정의합니다.

for irow in range(2):
    axs[irow, 0].yaxis.set_ticks(np.arange(0, 10, 2))

플롯에 제목 추가

전체 플롯에 제목을 추가합니다.

fig.suptitle('Colors in the default prop_cycle', fontsize='large')

플롯 표시

마지막으로, 플롯을 표시합니다.

plt.show()

요약

이 랩에서는 Python Matplotlib 을 사용하여 플롯을 만드는 방법을 배웠습니다. 색상 및 선 너비를 포함하여 플롯의 속성을 사용자 정의했습니다. 또한 서브플롯을 만들고 모양을 사용자 정의하는 방법도 배웠습니다.