Python Matplotlib 를 이용한 극좌표 그래프 생성

소개

이 랩에서는 Python Matplotlib 라이브러리를 사용하여 극좌표 그래프를 생성하고 주석을 다는 방법을 배웁니다. 극좌표 그래프는 극좌표를 사용하여 그려진 그래프입니다. 파동, 계절, 조수와 같은 주기적 현상을 시각화하는 데 유용합니다.

VM 팁

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

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

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

라이브러리 임포트

시작하기 위해 먼저 필요한 라이브러리를 임포트해야 합니다. 이 경우, numpy와 matplotlib가 필요합니다.

import matplotlib.pyplot as plt
import numpy as np

극좌표 그래프 생성

다음으로, figure 를 정의하고 극좌표 투영 (polar projection) 을 갖도록 지정하여 극좌표 그래프를 생성합니다. 또한 플로팅에 사용할 반경 (radius) 및 세타 (theta) 값을 정의합니다.

fig = plt.figure()
ax = fig.add_subplot(projection='polar')
r = np.arange(0, 1, 0.001)
theta = 2 * 2*np.pi * r
line, = ax.plot(theta, r, color='#ee8d18', lw=3)

주석 추가

극좌표 그래프에 주석을 추가하려면 주석의 위치를 지정해야 합니다. 이 경우, 그래프의 특정 지점을 선택하고 주석을 추가합니다.

ind = 800
thisr, thistheta = r[ind], theta[ind]
ax.plot([thistheta], [thisr], 'o')
ax.annotate('a polar annotation',
            xy=(thistheta, thisr),  ## theta, radius
            xytext=(0.05, 0.05),    ## fraction, fraction
            textcoords='figure fraction',
            arrowprops=dict(facecolor='black', shrink=0.05),
            horizontalalignment='left',
            verticalalignment='bottom',
            )

그래프 표시

이제 plt.show()를 사용하여 그래프를 표시할 수 있습니다.

plt.show()

요약

이 랩에서는 Python Matplotlib 라이브러리를 사용하여 극좌표 그래프를 생성하고 주석을 추가하는 방법을 배웠습니다. numpy를 사용하여 반지름 (radius) 과 세타 (theta) 값을 정의하고, plt.annotate()를 사용하여 그래프에 주석을 추가했습니다. 또한 plt.show()를 사용하여 그래프를 표시했습니다.