回答編集履歴
1
追記
test
CHANGED
@@ -11,3 +11,20 @@
|
|
11
11
|
zrow = [0] * n
|
12
12
|
zrow[i] = 1
|
13
13
|
```
|
14
|
+
**追記**
|
15
|
+
> n行n列の0行列を作ったあとで対角成分に1をセットするという順序で作りたいです。
|
16
|
+
|
17
|
+
この方針での回答はまだありませんね。
|
18
|
+
質問のコードの `return z` の前に 2行追加すればよいでしょう。
|
19
|
+
```Python
|
20
|
+
def identity_mat(n):
|
21
|
+
z = []
|
22
|
+
for i in range(n):
|
23
|
+
zrow = []
|
24
|
+
for j in range(n):
|
25
|
+
zrow.append(0)
|
26
|
+
z.append(zrow)
|
27
|
+
for i in range(n): # ★
|
28
|
+
z[i][i] = 1 # ★
|
29
|
+
return z
|
30
|
+
```
|