Python Numpy.pad 函数

Manav Narula 2023年1月3日 2021年3月21日
Python Numpy.pad 函数

在 Python 中,我们拥有 NumPy 模块来创建和使用数组。数组可以具有不同的大小和尺寸。填充是一种有用的方法,可用于补偿数组的大小。我们可以更改数组并添加一些填充值以更改其形状和大小。

我们还可以使用其他方法来重塑数组。尽管如此,此函数还是有好处的,因为它在使用后会自动调整数组内存的大小。

numpy.pad() 函数用于实现此目的。下面的代码显示了这个函数的一个例子。

import numpy as np
a = [1,2,3,4] 
b = np.pad(a,(3, 2), mode =  'constant', constant_values = (0, 5))
print(b)

输出:

[0 0 0 1 2 3 4 5 5]

在上面的示例中,第一个参数 (3,2) 元组指定在轴之前添加 3 个元素,并在轴的末端添加 2 个元素。

mode 参数指定填充数组时将使用哪种类型的值。在我们的代码中,我们使用常数 0 和 5 填充数组,但是我们可以将此模式更改为 medianmeanemptywrap 等不同类型。每种模式都提供不同的元素来填充数组。

我们也可以将此函数与多维数组一起使用。例如,

import numpy as np
a = np.array([[ 1.,  1.,  1.,  1.,  1.],
               [ 1.,  1.,  1.,  1.,  1.],
               [ 1.,  1.,  1.,  1.,  1.]])
b = np.pad(a, [(0, 1), (0, 1)], mode='constant')
print(b)

输出:

[[1. 1. 1. 1. 1. 0.]
 [1. 1. 1. 1. 1. 0.]
 [1. 1. 1. 1. 1. 0.]
 [0. 0. 0. 0. 0. 0.]]
注意
我们必须为二维数组指定两个元组,并且该函数的默认常数值为 0。
Author: Manav Narula
Manav Narula avatar Manav Narula avatar

Manav is a IT Professional who has a lot of experience as a core developer in many live projects. He is an avid learner who enjoys learning new things and sharing his findings whenever possible.

LinkedIn