NUMPY018 题解：下三角矩阵

题目

给定一个整数 $n$，创建 $n \times n$ 矩阵，对角线以下为 $1, 2, 3, ..., n-1$。

代码

import numpy as np

n = int(input())
Z = np.diag(1 + np.arange(n-1), k=-1)
print(Z)

代码详解

第一步：创建 [1, 2, 3, ..., n-1]

np.arange(n-1)  # [0, 1, 2, ..., n-2]
1 + np.arange(n-1)  # [1, 2, 3, ..., n-1]

第二步：创建对角矩阵

np.diag([1, 2, 3, 4], k=-1)

np.diag 参数说明

np.diag(数组, k)

k 的取值

其他实现方法

方法2：循环赋值

import numpy as np

n = int(input())
Z = np.zeros((n, n), dtype=int)
for i in range(1, n):
    Z[i, i-1] = i
print(Z)

核心知识点