3D 줄기 플롯

Beginner

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

소개

이 랩에서는 Python 의 Matplotlib 라이브러리를 사용하여 3D 줄기 플롯 (stem plot) 을 만드는 방법을 보여줍니다. 줄기 플롯은 기준선에서 데이터 지점까지 수직선을 그리고 끝에 마커를 배치하여 데이터 지점을 플로팅하는 방법입니다.

VM 팁

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

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

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

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

이 단계에서는 import 문을 사용하여 Matplotlib 및 Numpy 라이브러리를 가져오겠습니다.

import matplotlib.pyplot as plt
import numpy as np

데이터 정의

이 단계에서는 3D 줄기 플롯을 생성하는 데 사용할 데이터를 정의합니다. 각도에 대한 linspace 배열을 생성하고, 사인 및 코사인 함수를 사용하여 x 및 y 좌표를 계산합니다. 또한 z 좌표를 각도로 정의합니다.

theta = np.linspace(0, 2*np.pi)
x = np.cos(theta - np.pi/2)
y = np.sin(theta - np.pi/2)
z = theta

3D 줄기 플롯 생성

이 단계에서는 Matplotlib 의 stem 함수를 사용하여 3D 줄기 플롯을 생성합니다. stem 함수에 x, y, z 좌표를 인수로 전달합니다.

fig, ax = plt.subplots(subplot_kw=dict(projection='3d'))
ax.stem(x, y, z)

plt.show()

플롯 사용자 정의

이 단계에서는 bottom 매개변수를 사용하여 기준선을 변경하고, linefmt, markerfmt, 및 basefmt 매개변수를 사용하여 형식을 변경하여 3D 줄기 플롯을 사용자 정의합니다.

fig, ax = plt.subplots(subplot_kw=dict(projection='3d'))
markerline, stemlines, baseline = ax.stem(
    x, y, z, linefmt='grey', markerfmt='D', bottom=np.pi)
markerline.set_markerfacecolor('none')

plt.show()

플롯 방향 변경

이 단계에서는 orientation 매개변수를 사용하여 플롯의 방향을 변경합니다. 줄기가 x 방향으로 투영되고 기준선이 yz 평면에 있도록 방향을 'x'로 설정합니다.

fig, ax = plt.subplots(subplot_kw=dict(projection='3d'))
markerline, stemlines, baseline = ax.stem(x, y, z, bottom=-1, orientation='x')
ax.set(xlabel='x', ylabel='y', zlabel='z')

plt.show()

요약

이 랩에서는 Python 의 Matplotlib 라이브러리를 사용하여 3D 줄기 플롯을 만드는 방법을 배웠습니다. 데이터를 정의하는 것으로 시작하여 stem 함수를 사용하여 플롯을 생성했습니다. 또한 형식과 방향을 변경하여 플롯을 사용자 정의했습니다.