在plt.plot中使用label给出一个图例，而不是两个不同的图例

Question

Image having the Problem这是在同一行中显示标签，并且仅显示一个对象。

label=["Training Data","Linear Regression"]
plt.plot(X[:,1],X.dot(theta),label=label)
plt.legend()
plt.show()

Answer 1

是的，这是预期的。如果需要每行都有自己的标签，则需要分别绘制它们或创建自定义图例。后者可能如下所示：

import matplotlib.pyplot as plt
import numpy as np

X = np.random.randn(20,2)
theta = np.random.rand(2,2)

label=["Training Data","Linear Regression"]

lines = plt.plot(X[:,1], X.dot(theta))
plt.legend(lines, label)
plt.show()