替换 3d numpy 数组中的值

Question

有这两个 numpy 数组：

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

b = np.array([
    [
        [1,2],
        [2,3],
        [3,4]
    ],
    [
        [4,1],
        [5,2],
        [6,3]
    ]
])

有形状：

"a" shape: (2, 3, 5), "b" shape: (2, 3, 2)

我想将数组a 中的最后两个元素替换为数组b 中的元素，例如

c = np.array([
    [
        [1,2,3,1,2],
        [4,5,6,2,3],
        [7,8,9,3,4]
    ],
    [
        [1,3,5,4,1],
        [2,4,6,5,2],
        [1,1,1,6,3]
    ]
])

但是，np.hstack((a[:,:,:-2], b)) 会引发值错误：

除连接轴外的所有输入数组维度必须 完全匹配

通常看起来它不是正确使用的功能。附加也不起作用。

numpy 中是否有一种方法可以做到这一点，还是我需要使用 for 循环遍历数组并手动操作它们？

Answer 1

非覆盖方法：

• a[:,:,-2:] 获取末尾的零；使用a[:,:,:3]。

• 根据documentation，np.hstack(x) 等价于np.concatenate(x, axis=1)。由于您想在最里面的行加入矩阵，您应该使用axis=2。

代码：

>>> np.concatenate((a[:,:,:3], b), axis=2)
array([[[1, 2, 3, 1, 2],
        [4, 5, 6, 2, 3],
        [7, 8, 9, 3, 4]],

       [[1, 3, 5, 4, 1],
        [2, 4, 6, 5, 2],
        [1, 1, 1, 6, 3]]])

Answer 2

你可以像这样使用直接索引：

a[:, :, 3:] = b