Thêm một huyền thoại để PyPlot trong Matplotlib theo cách đơn giản nhất có thể

Question

TL; DR -> Làm thế nào người ta có thể tạo chú thích cho một đồ thị dòng trong Matplotlib 's PyPlot mà không tạo ra bất kỳ biến thêm?

Hãy xem xét kịch bản vẽ đồ thị dưới đây:

if __name__ == '__main__': 
    PyPlot.plot(total_lengths, sort_times_bubble, 'b-', 
       total_lengths, sort_times_ins, 'r-', 
       total_lengths, sort_times_merge_r, 'g+', 
       total_lengths, sort_times_merge_i, 'p-',) 
    PyPlot.title("Combined Statistics") 
    PyPlot.xlabel("Length of list (number)") 
    PyPlot.ylabel("Time taken (seconds)") 
    PyPlot.show() 

Như bạn có thể thấy, đây là một sử dụng rất cơ bản của matplotlib 's PyPlot. Đây tưởng tạo ra một đồ thị như hình dưới đây:

Không có gì đặc biệt, tôi biết. Tuy nhiên, không rõ dữ liệu đang được vẽ ở đâu (Tôi đang cố vẽ dữ liệu của một số thuật toán sắp xếp, độ dài so với thời gian đã thực hiện và tôi muốn đảm bảo mọi người biết được dòng nào). Vì vậy, tôi cần một huyền thoại, tuy nhiên, việc xem xét ví dụ sau đây (from the official site):

ax = subplot(1,1,1) 
p1, = ax.plot([1,2,3], label="line 1") 
p2, = ax.plot([3,2,1], label="line 2") 
p3, = ax.plot([2,3,1], label="line 3") 

handles, labels = ax.get_legend_handles_labels() 

# reverse the order 
ax.legend(handles[::-1], labels[::-1]) 

# or sort them by labels 
import operator 
hl = sorted(zip(handles, labels), 
      key=operator.itemgetter(1)) 
handles2, labels2 = zip(*hl) 

ax.legend(handles2, labels2) 

Bạn sẽ thấy rằng tôi cần phải tạo thêm một biến ax. Làm cách nào để thêm chú giải vào biểu đồ mà không có phải tạo biến phụ này và giữ lại sự đơn giản của tập lệnh hiện tại của tôi.

Answer 1

Thêm label= vào mỗi cuộc gọi plot() của bạn và sau đó gọi legend(loc='upper left').

xem xét mẫu này:

import numpy as np 
import pylab 
x = np.linspace(0, 20, 1000) 
y1 = np.sin(x) 
y2 = np.cos(x) 

pylab.plot(x, y1, '-b', label='sine') 
pylab.plot(x, y2, '-r', label='cosine') 
pylab.legend(loc='upper left') 
pylab.ylim(-1.5, 2.0) 
pylab.show() 

Hơi sửa đổi từ hướng dẫn này: http://jakevdp.github.io/mpl_tutorial/tutorial_pages/tut1.html

Answer 2

Thêm nhãn cho mỗi đối số trong cuộc gọi âm mưu của bạn tương ứng với loạt nó được vẽ đồ thị, ví dụ: label = "series 1"

Sau đó, chỉ cần thêm Pyplot.legend() để dưới cùng của kịch bản của bạn và truyền thuyết sẽ hiển thị các nhãn này.

Answer 3

Dưới đây là một ví dụ để giúp bạn ra ngoài ...

fig = plt.figure(figsize=(10,5)) 
ax = fig.add_subplot(111) 
ax.set_title('ADR vs Rating (CS:GO)') 
ax.scatter(x=data[:,0],y=data[:,1],label='Data') 
plt.plot(data[:,0], m*data[:,0] + b,color='red',label='Our Fitting 
Line') 
ax.set_xlabel('ADR') 
ax.set_ylabel('Rating') 
ax.legend(loc='best') 
plt.show() 

Answer 4

Bạn có thể truy cập cá thể Trục (ax) với plt.gca().Trong trường hợp này, bạn có thể sử dụng

plt.gca().legend() 

Bạn có thể làm điều này bằng cách sử dụng từ khóa label= trong mỗi plt.plot() cuộc gọi của bạn hoặc bằng cách gán nhãn của bạn như một tuple trong legend, như trong ví dụ làm việc này:

import numpy as np 
import matplotlib.pyplot as plt 
x = np.linspace(-0.75,1,100) 
y0 = np.exp(2 + 3*x - 7*x**3) 
y1 = 7-4*np.sin(4*x) 
plt.plot(x,y0,x,y1) 
plt.gca().legend(('y0','y1')) 
plt.show() 

Tuy nhiên, nếu bạn cần truy cập vào dụ Axes hơn một lần, tôi khuyên bạn nên lưu nó vào biến ax với

ax = plt.gca() 

và sau đó gọi ax thay vì plt.gca().