python - 从一组起点和终点绘制线段

Question

假设你有一个形状为(n,2,2)的数组a，它由n行的起点和终点的坐标组成分段网。应该如何绘制这些线段。我的代码不起作用，尽管它逐个元素起作用，但不是一次全部起作用。这是我的代码:

from matplotlib import pyplot as plt
import numpy as np
a=np.asarray([[[0,0],[1,1]],[[1,2],[2,3]],[[5,2],[2,6]],[[7,3],[5,9]]])
plt.plot(a[:,:,0],a[:,:,1])
plt.show()

我想得到的是: 但是我的代码输出是: 我应该如何纠正这个问题？

Answer 1

你只需要转置你的数组:

plt.plot(a[:,:,0].T,a[:,:,1].T)

原因是 matplotlib 对二维数组 a 轴的标准解释，其中 a[i,j] 的意思是是第 j 个图的第 i 个值。