Python | numpy.fill_diagonal() 方法

简介

numpy.fill_diagonal()方法用于将数组的对角线元素设置为给定的值。它是 NumPy 库中的一个功能强大的函数。

语法

numpy.fill_diagonal(arr, val, wrap=False)

• arr：输入数组
• val：要设置的值
• wrap：如果为真，则将对角线视为循环，并在填充后进行包装

返回值

这个方法没有任何返回值，而是直接修改了原始数组。

示例

让我们通过一个示例来理解numpy.fill_diagonal()的用法：

import numpy as np

# 创建一个3x3的数组
arr = np.zeros((3, 3), dtype=int)
print("原始数组：")
print(arr)

# 设置对角线元素为1
np.fill_diagonal(arr, 1)

print("修改后的数组：")
print(arr)

输出结果

原始数组：
[[0 0 0]
 [0 0 0]
 [0 0 0]]
修改后的数组：
[[1 0 0]
 [0 1 0]
 [0 0 1]]

解释

在上述示例中，我们首先使用numpy.zeros()函数创建一个3x3的零数组。然后，我们使用numpy.fill_diagonal()方法将对角线元素设置为1。这样，数组的对角线元素被成功地修改为所需值：1。

注意事项

• 参数val的类型必须与数组元素的类型兼容。否则，可能会引发类型不匹配的错误。
• 当设置wrap参数为 True 时，如果数组是具有一个以上维度的周期性（即在特定轴上重复），则该方法可以将设置对角线视为循环。
• 对于大型数组，numpy.fill_diagonal()方法的执行速度较快，因为它是使用了被编译为 C 的 NumPy 库来进行操作。

更多关于 NumPy 库的详细信息，请查阅 NumPy Documentation。