如何向NumPy矩阵中按指定seam位置插入元素并右移原有元素？

嘿，这个需求我之前在做图像seam扩展的时候刚好碰到过，用NumPy的矢量化操作就能搞定，比写循环快太多了！

核心思路 #

咱们要做的其实就是给原矩阵每列对应位置插入一个元素，同时把插入点右侧的元素都往右挪一位，最终得到宽一列的新矩阵。步骤很清晰：

• 先定位所有需要插入的位置：从seam矩阵里找出每列那个标记点的行、列索引
• 创建一个比原矩阵多一列的空结果矩阵
• 用NumPy的高级索引批量把原矩阵的元素放到新矩阵的对应位置，中间留出插入点放指定值

代码示例 #

先拿个简单的例子演示，假设原矩阵和seam矩阵是这样的：

import numpy as np

# 3x3的原矩阵
original = np.array([
    [10, 20, 30],
    [40, 50, 60],
    [70, 80, 90]
])

# seam矩阵：每列只有一个1，标记该行该列是插入位置
seam_mask = np.array([
    [0, 1, 0],
    [1, 0, 0],
    [0, 0, 1]
])

第一步，提取所有插入位置的坐标：

# 找出seam_mask中值为1的位置，得到(行索引数组, 列索引数组)
insert_rows, insert_cols = np.where(seam_mask == 1)

然后构建结果矩阵，这里我们插入0，你可以换成任何需要的数值（比如相邻元素的值）：

# 初始化结果矩阵，形状是(行数, 原列数+1)
result = np.zeros((original.shape[0], original.shape[1]+1), dtype=original.dtype)

# 把原矩阵每行的前insert_cols[i]个元素复制到结果矩阵的对应位置
result[insert_rows, :insert_cols[:, None]] = original[insert_rows, :insert_cols[:, None]]

# 在标记位置插入指定数值
result[insert_rows, insert_cols] = 0  # 这里可以改成你需要的，比如original[insert_rows, insert_cols]

# 把原矩阵每行insert_cols[i]及以后的元素复制到结果矩阵的insert_cols[i]+1及以后的位置
result[insert_rows, insert_cols[:, None]+1:] = original[insert_rows, insert_cols[:, None]:]

运行完你会得到这个结果：

[[10,  0, 20, 30],
 [ 0, 40, 50, 60],
 [70, 80,  0, 90]]

完全符合要求：每个标记点插入了0，原位置的元素都向右挪了一位！

小细节提示 #

• insert_cols[:, None]是把一维数组转成二维，让NumPy能正确进行广播索引，不然会出现维度不匹配的报错
• 如果你的seam矩阵不是用1标记，而是直接存插入位置的行号（比如每列的值就是对应的行索引），那提取坐标可以改成这样：

  insert_cols = np.arange(original.shape[1])
  insert_rows = seam_matrix[insert_cols, np.arange(original.shape[1])]
  

• 插入值可以灵活调整，比如要插入左边元素的值，就写成result[insert_rows, insert_cols] = original[insert_rows, insert_cols-1]（注意第一列的边界情况，不过你说输入满足预设条件，应该已经处理好了）

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