파이썬 Matplotlib에서 줄기 플롯(Stem Plot) 그리는 방법

Matplotlib에서 줄기 플롯(stem plot)을 그리려면 stem() 메서드를 사용하면 됩니다. 이 메서드는 기준선(baseline)에서 Y 좌표까지 수직선을 생성하고, 선 끝에 마커를 표시하여 데이터 값을 시각적으로 잘 드러내 줍니다.

구현 단계

• 그림 크기를 설정하고 서브플롯 주변 및 사이의 여백을 조정합니다.
• NumPy를 사용하여 x, y 데이터 포인트를 생성합니다.
• stem() 메서드로 줄기 플롯을 만듭니다.
• 마커의 면 색상을 빨간색으로 설정합니다.
• show() 메서드를 호출하여 그림을 화면에 표시합니다.

예제 코드

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

x = np.linspace(0.1, 2 * np.pi, 41)
y = np.exp(np.sin(x))

markerline, stemlines, baseline = plt.stem(x, y, linefmt='grey', markerfmt='*', bottom=1.1)

markerline.set_markerfacecolor('red')

plt.show()

실행 결과

위 코드를 실행하면 회색 줄기와 별(*) 모양 마커로 구성된 줄기 플롯이 출력되며, 각 마커의 내부는 빨간색으로 채워집니다. bottom=1.1 옵션으로 기준선이 y=1.1 위치에 그려지는 것도 확인할 수 있습니다.

주요 파라미터 설명

• linefmt: 줄기 선의 색상과 스타일을 지정합니다 (예: 'grey').
• markerfmt: 선 끝에 표시되는 마커의 형태를 지정합니다 (예: '*').
• bottom: 줄기가 시작되는 기준선의 y값을 지정합니다.

stem() 메서드는 markerline, stemlines, baseline 세 가지 객체를 반환하므로, 위 예제처럼 반환값을 받아 set_markerfacecolor() 등의 메서드로 세부 스타일을 자유롭게 커스터마이징할 수 있습니다.