Numpy 调整大小并填充特定值

Question

如何调整 numpy 数组的大小并用特定值填充它（如果扩展了某些维度）？

我找到了一种使用 np.pad 扩展数组的方法，但我无法缩短它：

>>> import numpy as np
>>> a = np.ndarray((5, 5), dtype=np.uint16)
>>> a
array([[0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0]], dtype=uint16)
>>> np.pad(a, ((0, 1), (0,3)), mode='constant', constant_values=9)
array([[0, 0, 0, 0, 0, 9, 9, 9],
       [0, 0, 0, 0, 0, 9, 9, 9],
       [0, 0, 0, 0, 0, 9, 9, 9],
       [0, 0, 0, 0, 0, 9, 9, 9],
       [0, 0, 0, 0, 0, 9, 9, 9],
       [9, 9, 9, 9, 9, 9, 9, 9]], dtype=uint16)

如果我使用 resize，我无法指定我想要使用的值。

>>> a.fill(5)
>>> a.resize((2, 7))
>>> a
array([[5, 5, 5, 5, 5, 5, 5],
       [5, 5, 5, 5, 5, 5, 5]], dtype=uint16)

但是我想要

>>> a
array([[5, 5, 5, 5, 5, 9, 9],
       [5, 5, 5, 5, 5, 9, 9]], dtype=uint16)

经过一些测试，我创建了这个函数，但它仅在您更改 x_value 或使用较低的 y_value 时起作用，如果您需要增加 y 维度，它不起作用，为什么？

VALUE_TO_FILL = 9
def resize(self, x_value, y_value):
    x_diff = self.np_array.shape[0] - x_value
    y_diff = self.np_array.shape[1] - y_value
    self.np_array.resize((x_value, y_value), refcheck=False)
    if x_diff < 0:
        self.np_array[x_diff:, :] = VALUE_TO_FILL
    if y_diff < 0:
        self.np_array[:, y_diff:] = VALUE_TO_FILL

Answer 1

要缩短它，您可以在切片中使用负值：

>>> import numpy as np
>>> a = np.ndarray((5, 5), dtype=np.uint16)
>>> a
array([[0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0]], dtype=uint16)
>>> b = a[0:-1,0:-3]
>>> b
array([[0, 0],
       [0, 0],
       [0, 0],
       [0, 0]], dtype=uint16)

Answer 2

您的阵列具有固定大小的数据缓冲区。您可以在不更改该缓冲区的情况下重塑数组。您可以在不更改缓冲区的情况下获取切片 (view)。但是你不能在不改变缓冲区的情况下向数组添加值。

一般来说 resize returns 是一个新的数组和一个新的数据缓冲区。

pad 是处理一般情况的复杂函数。但最简单的方法是创建 empty 目标数组，填充它，然后将输入复制到正确的位置。

或者 pad 可以创建填充数组并将它们与原始数组连接起来。但是 concatenate 也使空的 return 和副本。

带裁剪的自己动手垫可以构造为：

n,m = X.shape
R = np.empty((k,l))
R.fill(value)
<calc slices from n,m,k,l>
R[slice1] = X[slice2]

计算切片可能需要 if-else 测试或等效 min/max。你或许可以算出这些细节。

这可能是所有需要的

R[:X.shape[0],:X.shape[1]]=X[:R.shape[0],:R.shape[1]]

那是因为如果切片大于维度就没有问题。

In [37]: np.arange(5)[:10]
Out[37]: array([0, 1, 2, 3, 4])

因此，例如：

In [38]: X=np.ones((3,4),int)    
In [39]: R=np.empty((2,5),int)
In [40]: R.fill(9)

In [41]: R[:X.shape[0],:X.shape[1]]=X[:R.shape[0],:R.shape[1]]

In [42]: R
Out[42]: 
array([[1, 1, 1, 1, 9],
       [1, 1, 1, 1, 9]])

Numpy 调整大小并填充特定值

Numpy resize and fill with specific value

python

resize

numpy