arrays - 绘制选定的 numpy 数组行

Question

考虑一个小的 numpy 数组:

array([[  0.,   1.,   0.,   1.,   0.,   0.,   0.,   0.,   0.,   1.],
   [  0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.],
   [  0.,   0.,   0.,   0.,  18.,  15.,  25.,   0.,   0.,   0.],
   [  0.,   0.,   0.,  23.,  19.,  20.,  20.,   0.,   0.,   0.],
   [  0.,   0.,  20.,  22.,  26.,  23.,  18.,   0.,   0.,   0.],
   [  0.,   0.,   0.,  23.,  16.,  20.,  13.,   0.,   0.,   0.],
   [  0.,   0.,   0.,   0.,  18.,  20.,  18.,   0.,   0.,   0.],
   [  0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.],
   [  0.,   1.,   0.,   0.,   1.,   0.,   0.,   0.,   0.,   0.]])

我想绘制，比如说从第 3 行到第 6 行，即我的 numpy 数组的一部分(我来自 matlab 背景)。我怎么能循环这个？或者如何在同一个图中绘制多行我的 numpy 数组？

到目前为止，我已经尝试过；我定义了一个任意的 x:

x = np.arange(0,10)

那么如果我使用

plt.plot(x,data[3,:])

绘制第三行，效果很好。如果我尝试，就会出现问题:

plt.plot(x,data[3:4,:])

我收到错误“x 和 y 必须具有相同的第一个维度”，我理解这是因为他将第 3 行和第 4 行堆叠在一起，因此 x 和 y 的维度不同。我该如何克服呢？

谢谢

Answer 1

正如错误所暗示的，您的 data.shape = (1,10) 与您的输入 x.shape = (10,) 不一致。要解决这个问题，您可以使用 .T 转置数据，即

plt.plot(x, data[3:4,:].T)

另外，请记住 data[3:4,:] 与 data[3,:] 相同，您需要使用 data[3:5,:] 例如获取第 3 行和第 4 行。