如何在循环中使用np.insert实现向数组指定位置插入目标数组

你的问题核心在于每次插入操作会改变原数组长度，导致后续的插入位置参数失效——因为后续的pos值还是基于原数组的位置，而非插入后的新数组。下面提供两种可行的解决方案：

方法一：带偏移量的循环（直观易理解） #

每次插入后，记录已经插入的总元素数（偏移量），后续插入位置基于原pos值加上这个偏移量，就能保证插入到正确的位置：

import numpy as np

pos = np.array([1, 3, 4])
arr = np.array([0, 1, 2, 3, 4, 5])
vals = np.array([-1,-1])

offset = 0  # 记录已插入的总元素数
for p in pos:
    insert_pos = p + offset
    arr = np.insert(arr, insert_pos, vals)
    offset += len(vals)  # 每次插入len(vals)个元素，更新偏移量

print(arr)
# 输出：[ 0 -1 -1  1  2 -1 -1  3 -1 -1  4  5]

方法二：向量化实现（高效适合大数据） #

如果处理的数组规模较大，循环的效率会偏低，可以用numpy的矢量化操作一次性完成插入：

import numpy as np

pos = np.array([1, 3, 4])
arr = np.array([0, 1, 2, 3, 4, 5])
vals = np.array([-1,-1])

k = len(vals)  # 每个位置插入的元素数量
result_len = len(arr) + len(pos) * k
result = np.empty(result_len, dtype=arr.dtype)

# 填充原数组元素到正确位置
original_indices = np.arange(len(arr))
# 计算每个原索引前有多少个插入位置
insert_counts = np.searchsorted(pos, original_indices, side='right')
original_target = original_indices + insert_counts * k
result[original_target] = arr

# 填充插入元素到指定位置
insert_starts = pos + np.arange(len(pos)) * k
for start in insert_starts:
    result[start:start+k] = vals

print(result)
# 输出：[ 0 -1 -1  1  2 -1 -1  3 -1 -1  4  5]

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