如何使用Python实现numpy矩阵前一行内容复制到下一行

需求说明 #

需要实现矩阵行复制偏移效果：将矩阵每行内容复制到下一行位置，首行填充预设初始行M0，最终输出和原矩阵维度一致的新矩阵。

原有实现代码 #

import numpy as np
Total_T = 6
Total_P = 8

M0 = np.array([[0,0,0,0,0,1,1,1]])


M = np.random.randint(10,size = (Total_T, Total_P))
for k in range(0,Total_T):
    for l in range (0,Total_P):
        if k == Total_T-1:
            M[Total_T-1][l] = M0[0][l]
        #if k == 0:
            #M[0][l] = M0[0][l]
        else:
            M


M_prev = np.empty((Total_T, Total_P), dtype = object)
if M is not None:
  for k in range(0,Total_T):
    for l in range (0,Total_P):
        if k == 0:
            M_prev[0][l] = M0[0][l]
        else:
            M_prev = M[:].copy()

结果对比 #

• 代码实际生成的M矩阵：

array([[4, 8, 1, 6, 2, 4, 9, 6],
       [3, 9, 1, 6, 9, 2, 7, 1],
       [9, 4, 5, 9, 3, 5, 1, 2],
       [8, 3, 5, 2, 8, 5, 4, 8],
       [6, 3, 4, 3, 7, 9, 8, 4],
       [0, 0, 0, 0, 0, 1, 1, 1]])

• 预期M_prev输出：

array([[0, 0, 0, 0, 0, 1, 1, 1],
       [4, 8, 1, 6, 2, 4, 9, 6],
       [3, 9, 1, 6, 9, 2, 7, 1],
       [9, 4, 5, 9, 3, 5, 1, 2],
       [8, 3, 5, 2, 8, 5, 4, 8],
       [6, 3, 4, 3, 7, 9, 8, 4]])

• 原代码实际输出的M_prev：

array([[4, 8, 1, 6, 2, 4, 9, 6],
       [3, 9, 1, 6, 9, 2, 7, 1],
       [9, 4, 5, 9, 3, 5, 1, 2],
       [8, 3, 5, 2, 8, 5, 4, 8],
       [6, 3, 4, 3, 7, 9, 8, 4],
       [0, 0, 0, 0, 0, 1, 1, 1]])

错误原因 #

1. 生成M_prev的循环逻辑完全错误：遍历到k>=1的行时直接执行M_prev = M[:].copy()，每次都会把整个M完整复制给M_prev，覆盖掉之前给首行赋的M0值，最终得到的M_prev和M完全一致。
2. 逐元素双层遍历属于冗余写法，numpy原生支持整行/整列切片操作，不需要手动遍历每个元素，代码更简洁、运行效率更高。
3. 初始化M_prev时指定dtype=object没有必要，直接生成和M同数据类型的数组即可。

正确实现 #

直接使用numpy切片语法即可快速实现行复制偏移，不需要循环：

import numpy as np
Total_T = 6
Total_P = 8
M0 = np.array([[0,0,0,0,0,1,1,1]])

# 生成随机矩阵M，直接替换最后一行为M0，不需要逐元素循环
M = np.random.randint(10,size = (Total_T, Total_P))
M[-1] = M0

# 构造M_prev：首行放M0，后续行依次取M的第0行到倒数第二行
M_prev = np.empty_like(M)
M_prev[0] = M0
M_prev[1:] = M[:-1]

运行后输出的M_prev和预期完全一致。如果需要实现通用的「每行内容复制到下一行」逻辑，直接执行M[1:] = M[:-1]即可，首行可根据业务需求自行赋值。

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