python - 用零填充 1D NumPy 数组以形成 2D 数组

Question

我有一个 numpy 数组:

arr=np.array([0,1,0,0.5])

我需要如下形成一个新数组，使得每个零元素重复三次，并且每个非零元素都有 2 个前面的零，后跟非零数字。简而言之，每个元素都重复三次，零就这样，非零则有 2 前面的 0，然后是数字本身。如下:

([0,1,0,0.5])=0,0,0, [for index 0]
              0,0,1 [for index 1]
              0,0,0 [for index 2, which again has a zero] and
              0,0,0.5

最终输出应该是:

new_arr=[0,0,0,0,0,1,0,0,0,0,0,0.5]

np.repeat() 重复所有数组元素 n 次，但我不希望这样。这应该怎么做呢？感谢您的帮助。

Answer 1

快速 reshape ，然后调用 np.pad会做的:

np.pad(arr.reshape(-1, 1), ((0, 0), (2, 0)), 'constant')

输出:

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

你会想再次把它压平。只需调用 .reshape(-1, ) 即可完成。

>>> np.pad(arr.reshape(-1, 1), ((0, 0), (2, 0)), 'constant').reshape(-1, )
array([ 0. ,  0. ,  0. ,  0. ,  0. ,  1. ,  0. ,  0. ,  0. ,  0. ,  0. ,
    0.5])