Question

我在这里阅读了大多数相关问题，但是在这种情况下，我似乎无法弄清楚如何使用np.pad。也许这并不意味着这个特定问题。

假设我有一个Numpy数组列表。每个数组的长度相同，例如2.列表本身必须填充为例如5个数组，也可以转换为numpy数组。填充元素应为填充零的数组。例如

arr = [array([0, 1]), array([1, 0]), array([1, 1])]
expected_output = array([array([0, 1]), array([1, 0]), array([1, 1]), array([0, 0]), array([0, 0])])

以下似乎有效，但我认为必须有一种更好，更有效的方法。实际上，此过程运行数十万次（即使不是数百万次），因此速度很重要。也许使用np.pad？

import numpy as np

def pad_array(l, item_size, pad_size=5):
  s = len(l)

  if s < pad_size:
    zeros = np.zeros(item_size)
    for _ in range(pad_size-s):
      # not sure if I need a `copy` of zeros here?
      l.append(zeros)

  return np.array(l)

B = [np.array([0,1]), np.array([1,0]), np.array([1,1])]
AB = pad_array(B, 2)

print(AB)

Answer 1

您可以这样重写函数：

import numpy as np


def pad_array(l, item_size, pad_size=5):

    if pad_size < len(l):
        return np.array(l)

    s = len(l)
    res = np.zeros((pad_size, item_size))  # create an array of (item_size, pad_size)
    res[:s] = l  # set the first rows equal to the elements of l

    return res


B = [np.array([0, 1]), np.array([1, 0]), np.array([1, 1])]
AB = pad_array(B, 2)

print(AB)

输出

[[0. 1.]
 [1. 0.]
 [1. 1.]
 [0. 0.]
 [0. 0.]]

这个想法是创建一个零数组，然后用输入列表中的值填充第一行。

Answer 2

似乎您想以numpy的角度在轴0的末端填充零。所以您需要的是

output = numpy.pad(arr, ((0,2),(0,0)), 'constant')

技巧是pad_width参数，您需要将其指定为pad_width=((0,2),(0,0))才能获得预期的输出。这是告诉pad()在轴0的开头插入0填充，在轴0的末端插入2填充，并在轴1的开头插入0填充，在轴1的末端插入0填充。根据{{3}}

，{1}}是pad_width

((before_1, after_1), … (before_N, after_N))告诉mode='constant'填充参数pad()指定的值，该值默认为0。

给定大小的数字键盘零点

2 个答案: