#NUMPY018. 下三角矩阵

下三角矩阵

下三角矩阵

给定一个整数 nn,创建一个 n×nn \times n 的矩阵,对角线以下(不含对角线)的值为 1,2,3,,n11, 2, 3, \dots, n-1

输入格式

一行,一个整数 nn3n83 \leq n \leq 8

输出格式

输出 n×nn \times n 的 numpy 矩阵。

样例

5
[[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]]

提示

np.arange 常用参数

  • np.arange(n) —— 0 到 n-1
  • np.arange(start, stop) —— start 到 stop-1
  • np.arange(start, stop, step) —— 按步长生成

np.diag 常用参数

  • np.diag(数组) —— 创建对角矩阵(默认 k=0)
  • np.diag(数组, k) —— k=0 主对角线,k=-1 下方第一条,k=1 上方第一条