Python中的 numpy.ndarray.fill()

numpy.ndarray.fill()是numpy库中的一个函数，可用于将ndarray中所有元素设置为相同的值。

语法

numpy.ndarray.fill(value)

参数说明：

• value：需要设置的值。

返回值

None。原数组将被修改。

示例

import numpy as np

# 创建一个3*3的ndarray
arr = np.zeros((3,3))

# 打印原数组
print(arr)

# 将数组中所有元素设置为2
arr.fill(2)

# 打印修改后的数组
print(arr)

输出：

[[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]
[[2. 2. 2.]
 [2. 2. 2.]
 [2. 2. 2.]]

注意事项

1. value必须是数字类型，否则会抛出TypeError异常。

2. fill()方法会直接在原数组上修改，不会返回新的数组，因此要谨慎使用。