在Python中使用Numpy填充

问题描述 投票:0回答:1

采用一个3维数组。如下例所示,从最终位置填充实例。也就是说,我需要填充沿着数组边缘值镜像的发声的反射。

Array a:

array ([array([[3, 1, 4, 1],
   [5, 9, 2, 6],
   [5, 3, 5, 8]]),
   array([[9, 7, 9, 3],
   [2, 3, 8, 4]]),
   array([[6, 2, 6, 4],
   [3, 3, 8, 3],
   [2, 7, 9, 5],
   [0, 2, 8, 8]])], dtype=object)

dim1 = a.shape[0]    # n
dim2 = max([i.shape[0] for i in a]) # m
dim3 = a[0].shape[1] # k

填充后最终矩阵的尺寸:

结果= np.zeros((dim1,dim2,dim3))

输出:

      [[[3. 1. 4. 1.]
        [5. 9. 2. 6.]
        [5. 3. 5. 8.]
       [5. 3. 5. 8.]]
      [[9. 7. 9. 3.]
       [2. 3. 8. 4.]
       [2. 3. 8. 4.]
       [9. 7. 9. 3.]]

      [[6. 2. 6. 4.]
       [3. 3. 8. 3.]
       [2. 7. 9. 5.]
       [0. 2. 8. 8.]]]

如何使用numpy.pad获得输出?

numpy padding
1个回答
0
投票

技巧是使用pad_width参数,并在每个尺寸上填充特定的行数。在这种情况下,pad_width应该具有形状(2,2)。

for b in a:
    extra_lines = dim2 - b.shape[0]
    c = np.pad(b, pad_width=[[0, extra_lines], [0, 0]], mode='symmetric')
    print(c, '\n')

[[3 1 4 1]
 [5 9 2 6]
 [5 3 5 8]
 [5 3 5 8]] 

[[9 7 9 3]
 [2 3 8 4]
 [2 3 8 4]
 [9 7 9 3]] 

[[6 2 6 4]
 [3 3 8 3]
 [2 7 9 5]
 [0 2 8 8]] 
© www.soinside.com 2019 - 2024. All rights reserved.