Python NumPy 数组固定长度:使用 pad 函数补零
可以使用 numpy.pad 函数来实现固定长度并补0的操作。
例如,假设我们有一个长度为3的numpy数组:
import numpy as np
arr = np.array([1, 2, 3])
我们可以使用 numpy.pad 函数将其扩展到长度为5,并在末尾补0:
padded_arr = np.pad(arr, (0, 2), mode='constant')
print(padded_arr)
# 输出:[1 2 3 0 0]
这里的第二个参数 (0, 2) 表示在数组的前面不补0,在后面补2个0。mode='constant' 表示使用常量 0 来进行补齐。
原文地址: https://www.cveoy.top/t/topic/nq15 著作权归作者所有。请勿转载和采集!