그래프에 유한한 직선 추가하기

이전 포스팅에서 그래프를 그리는 기본적인 방법과 범례를 추가하는 방법에 대해서 배웠습니다.

https://jack-channel-python.tistory.com/entry/%EA%B7%B8%EB%9E%98%ED%94%84%EC%97%90-%EB%B2%94%EB%A1%80legend-%EC%B6%94%EA%B0%80%ED%95%98%EA%B8%B0

그래프에 범례(legend) 추가하기

 

이번시간에는 이전과 같은 그래프를 그리되, 여기다가 유한한 직선을 하나 추가로 그려보도록 하겠습니다.

y1(t) = x^2

y2(t)=x^2+1000

먼저, 위 두개의 그래프를 그려보도록 하겠습니다.

import numpy as np
import matplotlib.pyplot as plt

X = np.linspace(0, 100, 100)
Y1 = X**2
Y2 = X**2 + 1000

plt.plot(X,Y1)
plt.plot(X,Y2)

plt.show()

위와 같은 그래프가 그려지는 것을 확인할 수 있습니다.

여기까지는 기본이고, 이제 유한한 직선을 하나 그려볼텐데요!

두개의 좌표를 잊는 유한한 직선을 그려줄 것이기 떄문에, 아래 함수를 사용해줍니다.

plt.gca().add_line(plt.Line2D( (x좌표들),(y좌표들) )

 

저희는 (10,30)과 (1000,3000)를 각각 시작점과 끝점으로 갖는 직선을 추가해보도록 하겠습니다.

import numpy as np
import matplotlib.pyplot as plt

X = np.linspace(0, 100, 100)
Y1 = X**2
Y2 = X**2 + 1000

plt.plot(X,Y1)
plt.plot(X,Y2)

plt.gca().add_line(plt.Line2D((10,30),(1000,3000),color='b'))#두개의 좌표를 잊는 유한한 직선을 그려줍니다.

plt.show()

이전 그래프와 다르게 시작점과 끝점을 잊는 파란색의 유한한 직선이 그려지는 것을 확인할 수 있습니다.

참고 및 출처:

https://pyvisuall.tistory.com/57