numpy - 使用条件与 matplotlib 一起绘图

Question

我有一个二维数组，我需要绘制 x 和 y 列，但只能在 x 的特定范围内绘制。我知道如何使用索引绘图，但我需要指定 x 的值。我有几个这样的数组，所以我想找到一种方法来做到这一点，而不必单独查看每个数组。

这是一个例子:

array([[  4.40148390e+03,   1.13200000e+00],
       [  4.40248390e+03,   1.12200000e+00],
       [  4.40348440e+03,   1.11600000e+00],
       [  4.40448440e+03,   1.10600000e+00],
       [  4.40548490e+03,   1.09200000e+00],
       [  4.40648490e+03,   1.07700000e+00],
       [  4.40748540e+03,   1.08700000e+00],
       [  4.40848540e+03,   1.09400000e+00],
       [  4.40948580e+03,   1.10200000e+00],
       [  4.41048580e+03,   1.09500000e+00],
       [  4.41148630e+03,   1.12000000e+00]])

假设我只需要 4402 < x < 4410但我不知道索引。我可以放这样的东西吗:plot(x, y, where(4402 < x < 4410)) ？

我觉得这里明显缺少一些东西 :p

Answer 1

您可以使用 ma​​tplotlib 在 x 轴上设置限制，以便不显示数据系列中的所有点。但是，我们可以按如下方式过滤上面的 numpy 数组:

a = array([[ .... ]])
x = a[:,0]
y = a[:,1]

filter = (x>4402)&(x<4410)

plot(x[filter],y[filter])