可以使用 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 来进行补齐。

Python NumPy 数组固定长度:使用 pad 函数补零

原文地址: https://www.cveoy.top/t/topic/nq15 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录