matplotlib - 将任意行添加到 ipython 笔记本中的 matplotlib 图中

Question

我对 python/matplotlib 以及通过 ipython 笔记本使用它都相当陌生。我正在尝试向现有图形添加一些注释线，但我不知道如何在图形上呈现这些线。因此，例如，如果我绘制以下内容:

import numpy as np
np.random.seed(5)
x = arange(1, 101)
y = 20 + 3 * x + np.random.normal(0, 60, 100)
p =  plot(x, y, "o")

我得到以下图表:

那么如何添加一条从 (70,100) 到 (70,250) 的垂直线？从 (70,100) 到 (90,200) 的对角线怎么样？

我已经用 Line2D() 尝试了一些方法，结果除了我自己的困惑之外什么也没有。在R中，我只需使用segments()函数来添加线段。 matplotlib 中有等效的吗？

Answer 1

您可以通过向 plot 命令提供相应的数据(线段的边界)来直接绘制所需的线:

绘图([x1, x2], [y1, y2], color='k', linestyle='-', linewidth=2)

(当然你可以选择颜色、线宽、线型等)

根据您的示例:

import numpy as np
import matplotlib.pyplot as plt

np.random.seed(5)
x = np.arange(1, 101)
y = 20 + 3 * x + np.random.normal(0, 60, 100)
plt.plot(x, y, "o")


# draw vertical line from (70,100) to (70, 250)
plt.plot([70, 70], [100, 250], 'k-', lw=2)

# draw diagonal line from (70, 90) to (90, 200)
plt.plot([70, 90], [90, 200], 'k-')

plt.show()