我可以制作我的散点图，但我不能在同一张图中用 matplotlib 画线

Question

我是初学者，所以这可能是一个愚蠢的问题。如果我运行以下代码，我会看到 x 和 y 的散点图，但回归线 plt.plot(x, estimated_y, color="r", linewidth=3.0) 没有出现。我使用协方差矩阵估计了y。 x、y 和 estimated_y 都是 numpy 数组。如果我单独运行plt.plot(x, estimated_y, color="r", linewidth=3.0)，我会看到一个空图。

plt.figure()
plt.scatter(x,y)
plt.plot(x, estimated_y, color="r", linewidth=3.0) 
plt.show()
plt.xlabel("x")
plt.ylabel("y")

感谢大家的帮助！

Answer 1

您的代码看起来相当不错。我添加了一些随机数据来创建一个最小的可重现示例：

import matplotlib.pyplot as plt

# create dummy data
x = list(range(0,10))
y = list(range(10,0,-1))
estimated_y = [1]*10

plt.figure()
plt.scatter(x,y)
plt.plot(x, estimated_y, color="r", linewidth=3.0) 
plt.show()
# do not add anything to the axes after this command. The command forces to terminate the "rendering", which is why everything afterwards opens a new plot

输出看起来不错：

但是，您可以确保绘制到相同的轴（一个图形可能包含多个轴或“子图”）

# open a figure + create a single axis
fig, ax = plt.subplots()
ax.scatter(x,y)
ax.plot(x, estimated_y, color="r", linewidth=3.0) 
plt.show()

您从基础调用的任何绘图函数，例如matplotlib.pyplot.plot() 或 plt.plot() 因为您已将库的一部分重命名/导入为 plt，将绘制到当前活动的轴。这应该适用于您的情况。

好处是，使用轴时，您可以计算在轴上绘制的线数： len(ax.get_lines())

1

它是一个，因为scatter() 不会绘制“线”而是“点”......现在，如果你用你的数据调用它，它也应该返回1。如果是这样并且您看不到红线，则可能是您的数据包含NaNs 或者它的绘制超出了限制。如果返回0，则您已将其绘制到一些不同的（可能是不可见的或尚未绘制的）轴上。