対角成分以外を0にする行列

対角成分以外を0とするプログラムを作りたいです

import
1def trans(a):
2    n=a.shape[0]
3    for i in range(n):
4        for j in range(i,n):
5            tmp=a[i][j]
6            a[i][j]=0
7            a[j][i]=0
8            a[j][i]=tmp       
9a=np.array([[2,5,3],[1,6,4],[9,3,8]])
10print('original matrix');print(a)
11trans(a)
12print('transposed matrix');print(a)

この状態で実行すると
original matrix
[[2 5 3]
[1 6 4]
[9 3 8]]
transposed matrix
[[2 0 0]
[5 6 0]
[3 4 8]]　と表示されてしまい
求めている
[[2 0 0]
[0 6 0]
[0 0 8]] という形になりません。
また、出来ればfor文で作りたいです。

行動規範の内容に同意します

回答2件

ベストアンサー

iとjが異なるときに0をセットすればよいです。

Python
1def trans(a):
2    for i in range(a.shape[0]):
3        for j in range(a.shape[1]):
4            if i != j:
5                a[i][j] = 0

投稿2020/07/22 07:02

8524ba23

総合スコア38341

下記のような感じでしょうか。for文ではないですが…

python3
1import numpy as np
2A = np.array([[2,5,3],[1,6,4],[9,3,8]])
3print(A)
4"""
5[[2 5 3]
6 [1 6 4]
7 [9 3 8]]
8"""
9E = np.identity(3)
10print(E)
11"""
12[[1. 0. 0.]
13 [0. 1. 0.]
14 [0. 0. 1.]]
15"""
16AE = A*E
17print(AE)
18"""
19[[2. 0. 0.]
20 [0. 6. 0.]
21 [0. 0. 8.]]
22"""