如何采用python语言绘制一条_如何使用matplotlib绘制一条线？

我正在检查ax.axvline是如何工作的,我写了一个小功能,类似于它的一部分：

import matplotlib.pyplot as plt

import matplotlib.lines as mlines

def newline(p1, p2):

ax = plt.gca()

xmin, xmax = ax.get_xbound()

if(p2[0] == p1[0]):

xmin = xmax = p1[0]

ymin, ymax = ax.get_ybound()

else:

ymax = p1[1]+(p2[1]-p1[1])/(p2[0]-p1[0])*(xmax-p1[0])

ymin = p1[1]+(p2[1]-p1[1])/(p2[0]-p1[0])*(xmin-p1[0])

l = mlines.Line2D([xmin,xmax], [ymin,ymax])

ax.add_line(l)

return l

因此,如果您运行以下代码,您将意识到它是如何工作的.该线将跨越您的绘图的整个范围(独立于其大小),并且该线的创建不依赖于轴内的任何数据点,而是仅在您需要指定的两个固定点中.

import numpy as np

x = np.linspace(0,10)

y = x**2

p1 = [1,20]

p2 = [6,70]

plt.plot(x, y)

newline(p1,p2)

plt.show()