Python 使用 NumPy 库为数组添加零边界
你可以使用 NumPy 库中的 pad 函数来给一个存在的数组添加边界为 0。下面是一个示例代码:
import numpy as np
# 创建一个存在的数组
arr = np.array([[1, 2, 3],
[4, 5, 6]])
# 使用 pad 函数给数组添加边界为 0
padded_arr = np.pad(arr, pad_width=1, mode='constant', constant_values=0)
print(padded_arr)
输出结果为:
[[0 0 0 0 0]
[0 1 2 3 0]
[0 4 5 6 0]
[0 0 0 0 0]]
在上面的代码中,我们使用 np.pad 函数给数组 arr 的每个维度的边界添加了 1 个 0。pad_width 参数指定了在每个维度上添加的边界宽度,mode 参数指定了填充的方式,constant_values 参数指定了填充的值。
原文地址: https://www.cveoy.top/t/topic/piAs 著作权归作者所有。请勿转载和采集!