在具有（a，b，c）形狀的 numpy 數組的元素之間添加零

Question

我有一個這樣的 numpy 數組

array([[[1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.]],

       [[1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.]]])

形狀 = (2, 3, 5)

我想要一個 output 看起來像這樣

output = array([[[1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.],
         [1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.],
         [1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.]],

        [[1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.],
         [1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.],
         [1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.]]])

注意：要插入的零的數量可以根據給定的因子而變化，在這種情況下，因子是 k=3，插入是 (k-1)，這意味着將在數字之間插入兩個零。 還給出了這個 output 我想獲得初始輸入

Answer 1

您可以使用numpy.zeros初始化所需形狀的output數組，然后索引以填充值：

k = 3
shape = a.shape
output = np.zeros(shape[:-1]+((shape[-1]-1)*k+1,))

output[...,::k] = a

output：

array([[[1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.],
        [1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.],
        [1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.]],

       [[1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.],
        [1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.],
        [1., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 1.]]])