Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Matplotlib pyplot.plot()으로 매개변수화된 곡선 그리기


pyplot.plot()을 사용하여 매개변수화된 곡선(parametric curve)을 그리려면 아래 단계를 따르면 됩니다.

  • 그림(figure) 크기를 설정하고 서브플롯 사이와 주변의 여백(padding)을 조정합니다.
  • 샘플 개수를 나타내는 변수 N을 초기화합니다.
  • numpy를 사용하여 t, r, x, y 데이터 포인트를 생성합니다.
  • figure와 서브플롯 집합을 생성합니다.
  • plot() 메서드로 x, y 데이터 포인트를 플롯합니다.
  • 그림을 화면에 표시하려면 show() 메서드를 호출합니다.

동작 원리

이 예제에서는 극좌표 방정식 r = 0.5 + cos(t)를 매개변수 t에 대해 정의한 후, 이를 직교좌표(x = r·cos(t), y = r·sin(t))로 변환하여 곡선을 그립니다. 0부터 2π까지 400개(N=400)의 샘플을 생성하기 때문에 매끄러운 곡선이 완성됩니다.

예제 코드

import numpy as np
import matplotlib.pyplot as plt

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

N = 400
t = np.linspace(0, 2 * np.pi, N)
r = 0.5 + np.cos(t)
x, y = r * np.cos(t), r * np.sin(t)
fig, ax = plt.subplots()
ax.plot(x, y)

plt.show()

실행 결과

Matplotlib pyplot.plot()으로 매개변수화된 곡선 그리기