为什么matplotlib中的图例无法正确显示颜色？

Question

我有一个图，其中显示了3种不同的线图。因此，我要明确指定图例以显示3种颜色，每种图分别显示一种。下面是一个玩具示例：

import matplotlib.pyplot as plt

for i in range(1,20):
    if i%3==0 and i%9!=0:
        plt.plot(range(1,20),[i+3 for i in range(1,20)], c='b')
    elif i%9==0:
        plt.plot(range(1,20),[i+9 for i in range(1,20)], c='r')
    else:
        plt.plot(range(1,20),range(1,20), c='g')
plt.legend(['Multiples of 3 only', 'Multiples of 9', 'All the rest'])
plt.show()

但是图例无法正确显示颜色。为什么会这样以及如何解决？

Answer 1

已解决：

import matplotlib.pyplot as plt

my_labels = {"x1" : "Multiples of 3", "x2" : "Multiples of 9","x3":'All of the rest'}

for i in range(1,20):
    if i%3==0 and i%9!=0:
        plt.plot(range(1,20),[i+3 for i in range(1,20)], c='b', label = my_labels["x1"])
        my_labels["x1"] = "_nolegend_"
    elif i%9==0:
        plt.plot(range(1,20),[i+9 for i in range(1,20)], c='r', label = my_labels["x2"])
        my_labels["x2"] = "_nolegend_"
    else:
        plt.plot(range(1,20),[j for j in range(1,20)],c='g', label = my_labels["x3"])
        my_labels["x3"] = "_nolegend_"
plt.legend(loc="best") #
plt.show()

请参阅this链接中提供的doc链接，这将有助于回答答案。

Answer 2

我尝试了Rex5的答案；它可以在这个玩具示例中使用，但是在我的实际情节（如下）中，由于某种原因，它仍然产生了错误的图例。

相反，如Rex5提供的link中所建议的那样，以下解决方案有效（在玩具示例和我的实际情节中），并且也更简单：

for i in range(1,20):
    if i%3==0 and i%9!=0:
        a, = plt.plot(range(1,20),[i+3 for i in range(1,20)], c='b', label = my_labels["x1"])
    elif i%9==0:
        b, = plt.plot(range(1,20),[i+9 for i in range(1,20)], c='r', label = my_labels["x2"])
    else:
        c, = plt.plot(range(1,20),[j for j in range(1,20)],c='g', label = my_labels["x3"])
plt.legend([a, b, c], ["Multiples of 3", "Multiples of 9", "All of the rest"])
plt.show()