python - 如何选择两个 numpy 索引的范围？

Question

我有一个简单的 numpy 数组，我想创建一个单独的数组，每两个索引取两个元素
例如:

x = np.arange(0,20)

print(x)
[ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19]

我的目标是获得一系列

[2 3 6 7 10 11 14 15 18 19]

我该怎么做？我试过这个:

print(x[1:len(x)-1:2])

[ 1  3  5  7  9 11 13 15 17]

但我只得到所有其他索引。

Answer 1

您可以使用传统的 start:stop:step 简单地做到这一点。通过 reshape 数组，索引，然后将其展平，没有任何模数的约定。尝试这个 -

通过将其 reshape 为 (-1,2)，您可以创建双元组序列

然后，您只需从 1 开始并执行 2 次

最后你把它压平。

x.reshape(-1,2)[1::2].flatten()

array([ 2,  3,  6,  7, 10, 11, 14, 15, 18, 19])

这应该是 明显更快 而不是使用数学运算来检查每个值的方法，因为这只是 reshape 和索引。