如何优雅实现从NumPy数组末尾起始的子数组循环遍历？

假设我们有NumPy数组：

import numpy as np
a = np.array([0,1,2,3,4,5,6,7,8,9])

想要从数组末尾往前数n=6的位置开始，按长度m=2遍历子区间，预期输出：

[4 5]
[6 7]
[8 9]

原代码因最后一次迭代切片右边界为0，导致输出空数组：

n = 6
m = 2
for i in range(0, n, m):
    print(a[-n+i:-n+i+m])

输出：

[4 5]
[6 7]
[]

最优方案：用NumPy reshape直接拆分 #

这是最简洁优雅的方式，先截取数组最后n个元素，再将其重塑为每行m个元素的二维数组，之后直接遍历每行即可：

n = 6
m = 2
# 先取最后n个元素，再按每行m个拆分
sub_arrays = a[-n:].reshape(-1, m)
for sub in sub_arrays:
    print(sub)

执行后直接得到预期输出，无需处理边界情况。

方案二：调整循环范围 #

通过修改range的结束值，让切片右边界超过数组长度时自动被NumPy截断，避免空切片：

n = 6
m = 2
# 让循环结束值为n+m，确保最后一次切片右边界超出数组长度
for i in range(0, n + m, m):
    current_slice = a[-n+i:-n+i+m]
    # 可选：如果n不是m的整数倍，过滤空切片
    if current_slice.size > 0:
        print(current_slice)

或者已知n是m的整数倍时，直接按子区间数量循环：

n = 6
m = 2
for idx in range(n // m):
    print(a[-n + idx*m : -n + (idx+1)*m])

方案三：简化条件判断写法 #

如果一定要用原循环逻辑，可通过简洁的三元表达式处理边界：

n = 6
m = 2
for i in range(0, n, m):
    end = -n + i + m
    print(a[-n+i:] if end == 0 else a[-n+i:end])

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