python groubi设置对角线矩阵为0

可以使用 NumPy 库来创建对角线矩阵，并将对角线元素设置为0。以下是示例代码：

import numpy as np

# 创建一个 5x5 的对角线矩阵
matrix = np.eye(5)

# 将对角线元素设置为0
np.fill_diagonal(matrix, 0)

print(matrix)

输出：

[[0. 1. 1. 1. 1.]
 [1. 0. 1. 1. 1.]
 [1. 1. 0. 1. 1.]
 [1. 1. 1. 0. 1.]
 [1. 1. 1. 1. 0.]]

在代码中，`np.eye(5)` 创建了一个 5x5 的单位矩阵，然后使用 `np.fill_diagonal(matrix, 0)` 将对角线元素设置为0。