如何遍历ndarray并逐对处理连续行以生成结果矩阵？

嘿，这个需求用NumPy来处理超简单，而且效率比手动迭代高多了！我给你两种方案，优先推荐向量运算的方式，再给你手动迭代的版本供参考～

NumPy天生支持数组的切片和广播运算，完全不需要写循环，代码简洁还高效。具体步骤如下：

1. 先确保你的数据是NumPy数组（如果是普通列表，先转成数组：arr = np.array(your_original_list)）
2. 取数组中除了最后一行的所有行：arr[:-1]
3. 取数组中除了第一行的所有行：arr[1:]
4. 直接对这两个切片做元素级运算（比如你需要的减法），得到的就是所有相邻行的运算结果矩阵

举个实际例子：

import numpy as np

# 构造你给出的示例数组
arr = np.array([
    [1,2,3,4,5,6,7,8,9,10],
    [11,22,33,44,55,66,77,88,99,100],
    [111,222,333,444,55,666,777,888,999,1000]
])

# 计算相邻行的差值（当前行 - 下一行）
result = arr[:-1] - arr[1:]
print(result)

运行后输出的结果就是：

[ [ -10  -20  -30  -40  -50  -60  -70  -80  -90  -90]
  [-100 -200 -300 -400    0 -600 -700 -800 -900 -900] ]

完全符合你要的：第一行是第0行减第1行的结果，第二行是第1行减第2行的结果。

如果你想手动实现迭代逻辑，也可以写循环来完成，适合刚开始接触NumPy的同学理解过程：

import numpy as np

arr = np.array([
    [1,2,3,4,5,6,7,8,9,10],
    [11,22,33,44,55,66,77,88,99,100],
    [111,222,333,444,55,666,777,888,999,1000]
])

# 初始化空列表存储结果
result_list = []

# 遍历到倒数第二行（因为要取当前行和下一行）
for i in range(len(arr) - 1):
    current_row = arr[i]
    next_row = arr[i+1]
    # 对两行对应元素做减法
    diff_row = current_row - next_row
    result_list.append(diff_row)

# 转成NumPy数组
result = np.array(result_list)
print(result)

这个代码的输出和方案一完全一致，但如果你的数组行数很多（比如上万行），方案一的向量化运算速度会快很多，因为NumPy的底层是C实现的，避免了Python循环的性能开销。

另外，如果你的运算不是减法，只需要把运算符换成你需要的就行，比如加法+、乘法*，甚至自定义的元素级函数（可以用np.vectorize或者直接利用广播特性实现）。

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