목록의 플롯에 범례 추가

Question

데이터 세트에 범례를 추가하는 방법이 확실하지 않습니다. 몇 가지 온라인 가이드를 시도했지만 어떤 도움도없는 것 같습니다.

L_list = np.linspace(0, 500, 6) #kg 
plt.hold(True) 
theta = 84 
print("Begin simulation...") 
print() 
for i in L_list: 
    print("Mass = ", i, "kgs") 
    X, Y, t = FlightEnvelope(i, theta) 
    #T, M, t = FlightEnvelope(L, i) 
    plt.plot(X,Y) 



print() 
print("Simulation complete.") 
print() 
print("=============================================================") 
print("=============================================================") 
plt.ylabel("Height of Rocket (m)") 
plt.xlabel("Horizontal distance traveled (m)") 
plt.title("Trajectory of Rocket") 
plt.grid(True) 
plt.show() 

Answer 1

그냥 사용

plt.legend(["A", "B", "C"])