CQUPT Python numpy 矩阵

【问题描述】

在数组主对角线上创建一个值为1,2,3,4的4*4矩阵，图例如下：

[[1. 0. 0. 0.]

 [0. 2. 0. 0.]

 [0. 0. 3. 0.]

 [0. 0. 0. 4.]]

【输入形式】

4
【输出形式】

[[1. 0. 0. 0.]

 [0. 2. 0. 0.]

 [0. 0. 3. 0.]

 [0. 0. 0. 4.]]

【完整代码】

import numpy as np

n = int(input())
arr = np.linspace(1, n, n)
# arr = np.arange(1, n + 1, 1.0)
print(np.diag(arr))

【代码讲解】