python - 如何在 Python 中用间隔索引数组的多个项目

Question

假设我有一个列表:

import numpy as np
a = [2, 4, 6, 8, ..., 1000] # total 500 elements
b = np.array(a)             # numpy version

我想获取第 1 到第 100、第 201 到第 300、第 401 到第 500 个元素并将它们放入一个新数组中。

为此，我尝试了以下代码:

a_sub = a[0:100] + a[200:300] + a[400:500]
b_sub = np.concatenate((b[0:100], b[200:300], b[400:500]))

但我想用一个简单的单行索引来做到这一点

说:

a_sub = a[(0:100, 200:300, 400:500)]
a_sub = a[[0:100, 200:300, 400:500]]
b_sub = b[[0:100, 200:300, 400:500]]
b_sub = b[[0:100, 200:300, 400:500]]

但是以上都是无效的，我找不到这样的单行索引。

Answer 1

您可以使用整形 np.reshape和 slicing , 像这样 -

np.array(a).reshape(-1,100)[::2].ravel()

如果 a 是一个 NumPy 数组，你可以这样做 -

a.reshape(-1,100)[::2].ravel()