如何將 python matplotlib.pyplot 圖例標記更改為序列號，如 1、2、3，而不是形狀或字符？

Question

import matplotlib.pyplot

plt.figure() 
plt.plot(x, 'r+', label='one')
plt.plot(x1, 'go--', label ='two')
plt.plot(y, 'ro', label='Three')
plt.legend()

在上面的代碼中，圖例標記是“r+”、“go--”和“ro”，但我希望它變成 1,2 和 3，因為有 3 個地塊。？ 誰能幫我解決這個問題？ 還有什么可以在不對數字進行硬編碼的情況下完成的嗎？ “““ 謝謝。

Answer 1

您可以使用生成器（例如itertools.count ）和next ：

import matplotlib.pyplot

x=x1=y=(0,0) # dummy data

markers = iter(['r+', 'go--', 'ro'])

plt.figure() 
plt.plot(x, next(markers), label='1')
plt.plot(x1, next(markers), label='2')
plt.plot(y, next(markers), label='3')
plt.legend()

output：