pythonでのforを用いた繰返し計算

前提・実現したいこと

プログラミング、python初心者でとても単純な質問かもしれませんが、お答えいただけると助かります。

二つのマトリックス同士を計算して、ある一定の数を超えると1で返し、それ以外は0となるようなバイナリの行列を作成したいと考えております。
しかし、繰返しで計算が行われなく、なにが間違っているのかを知りたいです。
よろしくおねがいいたします。

該当のソースコード

python
1matrix_list = [[0, 1, 0, 1, 0],
2               [1, 0, 1, 0, 1],
3               [0, 1, 0, 1, 0],
4               [1, 0, 1, 0, 1],
5               [0, 1, 0, 1, 0],]
6pop_matrix_list = [[200, 3000, 4000, 200, 1000],
7                   [5000, 3000, 20, 4000, 4000],
8                   [200, 3000, 4000, 200, 1000],
9                   [5000, 3000, 20, 4000, 4000],
10                   [200, 3000, 4000, 200, 1000],]
11
12def new_matrix():
13    for i in range(5):
14        for j in range(5):
15            cal = matrix_list[i][j] * pop_matrix_list[i][j] 
16            if cal >= 3000:
17                return 1
18            else:
19                return 0

出力結果

Python
1new_matrix()
20

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

回答3件

numpy使わないなら・・・

python
1matrix_list = [[0, 1, 0, 1, 0],
2               [1, 0, 1, 0, 1],
3               [0, 1, 0, 1, 0],
4               [1, 0, 1, 0, 1],
5               [0, 1, 0, 1, 0],]
6pop_matrix_list = [[200, 3000, 4000, 200, 1000],
7                   [5000, 3000, 20, 4000, 4000],
8                   [200, 3000, 4000, 200, 1000],
9                   [5000, 3000, 20, 4000, 4000],
10                   [200, 3000, 4000, 200, 1000],]
11
12def new_matrix(m1, m2, threashold):
13    return [[int(c1 * c2 >= threashold) for c1, c2 in zip(r1, r2)]
14            for r1, r2 in zip(m1, m2)]
15
16print(*new_matrix(matrix_list, pop_matrix_list, 3000), sep='\n')

投稿2019/12/25 08:00

編集2019/12/25 08:08

shiracamus

総合スコア5406

こういう事？

Python
1from pprint import pprint
2
3
4matrix_list = [[0, 1, 0, 1, 0],
5               [1, 0, 1, 0, 1],
6               [0, 1, 0, 1, 0],
7               [1, 0, 1, 0, 1],
8               [0, 1, 0, 1, 0],]
9pop_matrix_list = [[200, 3000, 4000, 200, 1000],
10                   [5000, 3000, 20, 4000, 4000],
11                   [200, 3000, 4000, 200, 1000],
12                   [5000, 3000, 20, 4000, 4000],
13                   [200, 3000, 4000, 200, 1000],]
14
15def new_matrix():
16    result = [[0] * 5 for _ in range(5)]
17    for i in range(5):
18        for j in range(5):
19            cal = matrix_list[i][j] * pop_matrix_list[i][j]
20            if cal >= 3000:
21                result[i][j] = 1
22    return result
23
24pprint(new_matrix())
25#[[0, 1, 0, 0, 0],
26# [1, 0, 0, 0, 1],
27# [0, 1, 0, 0, 0],
28# [1, 0, 0, 0, 1],
29# [0, 1, 0, 0, 0]]

投稿2019/12/25 07:57

magichan

総合スコア15898

退会済みユーザー

2019/12/25 12:56

なるほどです！ありがとうございます。

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

ベストアンサー

numpyを使えば簡単に実装できると思います。

python
1import numpy as np
2
3matrix_list = [[0, 1, 0, 1, 0],
4               [1, 0, 1, 0, 1],
5               [0, 1, 0, 1, 0],
6               [1, 0, 1, 0, 1],
7               [0, 1, 0, 1, 0],]
8pop_matrix_list = [[200, 3000, 4000, 200, 1000],
9                   [5000, 3000, 20, 4000, 4000],
10                   [200, 3000, 4000, 200, 1000],
11                   [5000, 3000, 20, 4000, 4000],
12                   [200, 3000, 4000, 200, 1000],]
13
14matrix = np.array(matrix_list)
15pop_matrix = np.array(pop_matrix_list)
16
17res = matrix * pop_matrix
18
19np.where(res > 3000,1, 0)

出力例

array([[0, 0, 0, 0, 0],
       [1, 0, 0, 0, 1],
       [0, 0, 0, 0, 0],
       [1, 0, 0, 0, 1],
       [0, 0, 0, 0, 0]])

投稿2019/12/25 07:53

編集2019/12/25 07:54

bamboo-nova

総合スコア1408

退会済みユーザー

2019/12/25 12:56

numpy を使うとかなり楽ですね。ありがとうございます。

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

あなたの回答

tips

プレビュー

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

質問の解決につながる回答をしましょう。サンプルコードなど、より具体的な説明があると質問者の理解の助けになります。また、読む側のことを考えた、分かりやすい文章を心がけましょう。

15分調べてもわからないことは
teratailで質問しよう！

ただいまの回答率
85.48%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう！