python N*Ｎの2次元配列を作りたい

Question

キーボードから整数値のSeedとNを入力して二次元配列を作りたいです。

コード
import numpy as np
Seed , N = [int(x) for x in input().split()]
np.random.seed(seed=Seed) # 乱数seedをSeedに設定
hairetu = np.random.randint(100 , size=N*N)
print(hairetu)

Seedに0、Nに2を入力すると出力結果が
[44 47 64 67]となります。
これを2次元配列として
[[44 47]
[64 67]]
となるように生成するにはどうしたらよいでしょうか。

Answer 1

hairetu = np.random.randint(100 , size=N*N).reshape(N, N)
という形にreshapeを使います。

Comments

[magepi]

試してみたらできました！ありがとうございます！