回答編集履歴
1
コメント欄の質問の回答を追加
answer
CHANGED
@@ -39,4 +39,30 @@
|
|
39
39
|
#[[1 2 3]
|
40
40
|
# [4 5 6]
|
41
41
|
# [7 8 9]]
|
42
|
+
```
|
43
|
+
|
44
|
+
---
|
45
|
+
**【追記】**
|
46
|
+
|
47
|
+
nxn を作る関数に改造する方法
|
48
|
+
|
49
|
+
- 入力値として `n` を受け取るように変更
|
50
|
+
- `np.array([1,2,3,4,5,6,7,8,9])` の箇所を `1~(n*n)` の配列を作成するように変更(`np.arange()`を使用する)
|
51
|
+
|
52
|
+
で良いかと思います。あとは呼び出し元にて引数付きで関数を呼ぶと動作確認できます。
|
53
|
+
|
54
|
+
```Python
|
55
|
+
import numpy as np
|
56
|
+
|
57
|
+
def tasks(n):
|
58
|
+
A = np.arange(1, n*n+1)
|
59
|
+
A = A.reshape((n,n))
|
60
|
+
return A
|
61
|
+
|
62
|
+
result = tasks(4)
|
63
|
+
print(result)
|
64
|
+
#[[ 1 2 3 4]
|
65
|
+
# [ 5 6 7 8]
|
66
|
+
# [ 9 10 11 12]
|
67
|
+
# [13 14 15 16]]
|
42
68
|
```
|