创建n×n矩阵

给定一个整数 $n$，创建一个 $n \times n$ 的矩阵，值从 $0$ 到 $n^2-1$。

输入格式

一行，一个整数 $n$（$3 \leq n \leq 5$）

输出格式

输出 $n \times n$ 的 numpy 矩阵。

样例

3

[[0 1 2]
 [3 4 5]
 [6 7 8]]

提示

• np.arange(n * n) —— 创建 $0$ 到 $n^2-1$ 的数组
• .reshape(n, n) —— 重塑为 $n \times n$ 矩阵