Python NumPy 创建二维数组边界为1,其余为0
你可以使用 NumPy 库中的 zeros 函数来创建一个指定大小的全 0 数组,然后将边界上的元素设置为 1。以下是一个示例代码:
import numpy as np
# 创建一个 3x3 的全 0 数组
arr = np.zeros((3, 3))
# 将边界上的元素设置为 1
arr[0, :] = 1
arr[-1, :] = 1
arr[:, 0] = 1
arr[:, -1] = 1
print(arr)
运行结果为:
[[1. 1. 1.]
[1. 0. 1.]
[1. 1. 1.]]
这样就创建了一个 3x3 的二维数组,边界上的元素为 1,其余元素为 0。你可以根据需要修改数组的大小。
原文地址: https://www.cveoy.top/t/topic/piAr 著作权归作者所有。请勿转载和采集!