在具有（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.]]])