Python NumPy 数组填充:固定长度不足补零
可以使用 numpy.pad() 函数来实现这个功能。该函数可以在数组的边缘添加一定数量的填充值,以达到指定的形状。以下是一个示例:
import numpy as np
a = np.array([1, 2, 3])
b = np.pad(a, (0, 5-len(a)), 'constant', constant_values=0)
print(b) # 输出 [1 2 3 0 0 0]
在上面的代码中,我们首先创建了一个长度为3的一维数组 a,然后使用 numpy.pad() 函数将其填充为长度为5的一维数组 b。函数的第一个参数是要填充的数组,第二个参数是一个元组,分别表示在数组的前面和后面要添加的填充元素的数量。在本例中,我们将前面添加了0个元素,后面添加了 5-3=2 个元素。第三个参数是填充方式,这里我们选择了 'constant' 表示使用常数填充。最后一个参数 constant_values 表示要使用的常数值,这里我们选择了 0。
通过这种方式,我们可以轻松地将一个固定长度的 NumPy 数组填充为指定长度,并且可以自定义填充的值。
原文地址: https://www.cveoy.top/t/topic/nq13 著作权归作者所有。请勿转载和采集!