본문 바로가기
Matplotlib/화살표

[파이썬 matplotlib] 그래프에 화살표 추가하기

by 만다린망고 2020. 11. 18.
반응형

[파이썬 matplotlib] 

그래프에 화살표 추가하기


annotation 의 목적으로 사용되는 화살표입니다. annotate 함수를 사용합니다. 설정하는 옵션들은 아래와 같습니다.


ha : horizontal alignment

va : vetical alignment

xytext : 텍스트 위치 

xy : 화살표 위치

arrowprops : 화살표 속성들


화살표 속성에는 아래와 같은 것들이 있습니다. 딕셔너리 형태로 입력해야 합니다.


facecolor : 채우기 색

edgecolor : 테두리색

shrink : 화살표 길이 

alpha : 투명도

arrowstyle : 화살표 스타일


화살표 스타일에는 아래와 같은 것들이 있습니다.


simple

fancy

wedge

-

->

-[

-|>

<-

<->

<|-

<|-|>

]-

]-[

|-|


아래는 예시입니다. 


import numpy as np

import matplotlib.pyplot as plt


X = np.linspace(0, 100, 100)

Y = X**2



plt.annotate('here',

ha = 'center', va = 'bottom',

xytext = (40, 4000),

xy = (20, 2000),

arrowprops = { 'facecolor' : 'r', 

              'edgecolor':'b', 

              'shrink' : 0.2, 

              'alpha':0.5

              })



plt.plot(X, Y)

plt.show()



아래는 스타일을 설정한 경우입니다. 특정 스타일의 경우 shrink 옵션을 없애주어야 합니다. 


import numpy as np

import matplotlib.pyplot as plt


X = np.linspace(0, 100, 100)

Y = X**2



plt.annotate('here',

ha = 'center', va = 'bottom',

xytext = (40, 4000),

xy = (20, 2000),

arrowprops = {

              'edgecolor':'b', 

              'alpha':0.5,

              'arrowstyle':'<->'

              })



plt.plot(X, Y)

plt.show()




반응형

댓글