Python 使用 NumPy 创建 5x5 矩阵并设置对角线下方值

可以使用 NumPy 的 fill_diagonal 函数来创建一个对角矩阵，并设置值 1, 2, 3, 4 落在其对角线下方位置。具体的步骤如下：

import numpy as np

# 创建一个 5*5 的零矩阵
matrix = np.zeros((5, 5))

# 设置值 1, 2, 3, 4 落在对角线下方位置
np.fill_diagonal(matrix[1:], [1, 2, 3, 4])

print(matrix)

输出结果为：

[[0. 0. 0. 0. 0.]
 [1. 0. 0. 0. 0.]
 [0. 2. 0. 0. 0.]
 [0. 0. 3. 0. 0.]
 [0. 0. 0. 4. 0.]]

可以看到，1, 2, 3, 4 分别位于对角线下方的位置。