如何用Numpy一行代码实现提取x个元素后跳过y个的数组切片？

针对你需要的提取x个元素、跳过y个元素的通用需求，有两种简洁的Numpy实现方式，不用手动组合多个切片：

方法一：重塑数组+切片（适用于数组长度为x+y整数倍的场景） #

如果你的数组长度刚好能被x+y整除，可以直接通过reshape将数组分组，再提取每组的前x个元素后展平：

import numpy as np
a = np.arange(10)
x, y = 2, 2

# 把数组拆成每组包含x+y个元素的二维数组
grouped = a.reshape(-1, x + y)
# 提取每组前x个元素，再展平为一维数组
result = grouped[:, :x].flatten()
# 输出：array([0, 1, 4, 5, 8, 9])

原理：reshape(-1, x+y)会自动计算组数，把原数组分成若干个长度为x+y的子数组；[:, :x]取每个子数组的前x个元素；最后flatten()把二维结果转回一维。

方法二：生成索引数组（通用场景，支持数组长度非整数倍） #

如果数组长度不能被x+y整除，用索引生成的方式更稳妥，能自动忽略最后一组不足x个的元素：

import numpy as np
a = np.arange(10)
x, y = 2, 2

# 生成所有符合规则的索引
indices = np.concatenate([np.arange(i, i+x) for i in range(0, len(a), x+y)])
# 过滤掉超出数组长度的索引（避免越界）
indices = indices[indices < len(a)]
result = a[indices]
# 输出：array([0, 1, 4, 5, 8, 9])

原理：range(0, len(a), x+y)生成每个分组的起始位置，np.arange(i, i+x)生成该分组内要提取的x个索引，最后用concatenate把所有索引拼起来，再用数组索引提取元素。

补充：你之前尝试的切片问题 #

a[:2:2]返回[0]是因为Numpy切片的格式是[start:stop:step]，这个切片的意思是从0开始，到索引2（不包含）结束，步长为2，所以只会取到索引0的元素，和你需要的分组提取逻辑完全不同。

内容的提问来源于stack exchange，提问作者soubaboy