[파이썬 matplotlib] 그래프에 도형 추가하기

[파이썬 matplotlib] 

그래프에 도형 추가하기

그래프에 그리드를 추가할 때는 patches.도형이름 함수를 사용합니다. 아래와 같은 함수들이 있습니다.

matplotlib.patches.Circle( (중심좌표), radius=반지름)

matplotlib.patches.Rectangle( (좌측하단모서리좌표), 가로길이,세로길이)

matplotlib.patches.Ellipse( (중심좌표), 가로,세로, angle=기울임)

위부터 원, 직사각형, 타원입니다. 아래와 같이 gca 를 이용하여 출력해야합니다. 

matplotlib.pyplot.gca().add_patch(여기에 입력) 

아래는 예시입니다.

import numpy as np

import matplotlib.patches as patches

import matplotlib.pyplot as plt

X=np.array([1,10])

Y=np.array([1,10])

plt.plot(X,Y,color='None')

#circle

shp=patches.Circle((2,2), radius=1, color='r')

plt.gca().add_patch(shp)

#rectangle

shp=patches.Rectangle((2,4), 3,1, color='b')

plt.gca().add_patch(shp)

#ellipse

shp=patches.Ellipse((6,6), 4,1, color='g',angle=45)

plt.gca().add_patch(shp)

plt.show()

원이 타원처럼 보이죠? 축의 길이의 비율을 실제 비율로 설정해봅시다. plt.axis('scaled') 를 추가해주시면 됩니다.