如何为numpy嵌套数组的每个子数组添加前置零？

针对你需要给NumPy二维数组的每个子数组（行）添加前置零的需求，这里提供几种高效的实现方法：

方法一：使用np.hstack拼接零数组 #

先创建一个与原数组行数相同、列数为1的全零数组，再与原数组水平拼接：

import numpy as np
a = np.array([[1, 2], [3, 4], [5, 6]])
# 生成对应形状的全零列
zeros_col = np.zeros((a.shape[0], 1), dtype=a.dtype)
# 水平拼接得到结果
b = np.hstack([zeros_col, a])

输出结果b即为[[0 1 2] [0 3 4] [0 5 6]]

方法二：使用np.insert批量插入（最简洁） #

直接指定axis=1（沿列方向）在每行开头插入0：

import numpy as np
a = np.array([[1, 2], [3, 4], [5, 6]])
b = np.insert(a, 0, 0, axis=1)

该方法无需额外创建零数组，一步完成插入操作。

关于你原循环方法无效的原因 #

你之前的循环代码：

for item in a:
    item = np.insert(item, 0, 0)

这里的item是原数组行的视图，但item = np.insert(...)只是将变量item指向了新创建的数组，并没有修改原数组的内容。如果一定要用循环实现，需要通过索引修改原数组：

import numpy as np
a = np.array([[1, 2], [3, 4], [5, 6]])
for i in range(a.shape[0]):
    a[i] = np.insert(a[i], 0, 0)

不过这种循环方式效率远低于NumPy的矢量操作，不推荐使用。

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