Matplotlib 를 이용한 방향 화살표가 있는 타원 그리기

Beginner

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

소개

이 랩에서는 Matplotlib 을 사용하여 방향 화살표가 있는 타원을 그리는 방법을 배우게 됩니다. 타원은 데이터를 표현하기 위해 데이터 시각화에서 일반적으로 사용되는 도형의 한 유형입니다. 타원에 방향 화살표를 추가하면 데이터의 방향에 대한 추가 정보를 제공할 수 있습니다.

VM 팁

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

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

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

Matplotlib 가져오기 및 Figure 와 Axis 생성

먼저, Matplotlib 을 가져오고 figure 와 axis 를 생성해야 합니다. Figure 와 axis 는 플롯을 위한 컨테이너입니다.

import matplotlib.pyplot as plt

## Create a figure and axis
fig, ax = plt.subplots(subplot_kw={"aspect": "equal"})

타원 생성

다음으로, Ellipse 클래스를 사용하여 타원을 생성해야 합니다. 타원의 중심, 타원의 너비와 높이, 회전 각도를 지정할 수 있습니다.

from matplotlib.patches import Ellipse

ellipse = Ellipse(
    xy=(2, 4),
    width=30,
    height=20,
    angle=35,
    facecolor="none",
    edgecolor="b"
)
ax.add_patch(ellipse)

방향 화살표 추가

타원의 단축 끝점에 마커를 플로팅하여 타원에 방향 화살표를 추가할 수 있습니다. get_co_vertices() 메서드를 사용하여 타원의 꼭지점 좌표를 얻을 수 있습니다. 그런 다음, Affine2D() 클래스를 사용하여 마커를 타원의 각도에 맞게 회전시킬 수 있습니다.

from matplotlib.markers import MarkerStyle
from matplotlib.transforms import Affine2D

## Plot an arrow marker at the end point of minor axis
vertices = ellipse.get_co_vertices()
t = Affine2D().rotate_deg(ellipse.angle)
ax.plot(
    vertices[0][0],
    vertices[0][1],
    color="b",
    marker=MarkerStyle(">", "full", t),
    markersize=10
)

방향 화살표 반전

방향 화살표를 반전시키려면 마커 유형을 >에서 <로 변경할 수 있습니다.

## To reverse the orientation arrow, switch the marker type from > to <.
ax.plot(
    vertices[0][0],
    vertices[0][1],
    color="b",
    marker=MarkerStyle("<", "full", t),
    markersize=10
)

플롯 표시

마지막으로, show() 메서드를 사용하여 플롯을 표시할 수 있습니다.

plt.show()

요약

축하합니다! Matplotlib 을 사용하여 방향 화살표가 있는 타원을 그리는 방법을 배웠습니다. 이 기술은 방향성을 가진 데이터를 시각화하는 데 유용할 수 있습니다.