python3の標準入力から取得したリストのiterationについて

Question

初めての質問になります。ご無礼なことあるかもしれませんが何卒よろしくお願います。

前提・実現したいこと

python3での実装です。　　　　　　　
2つの数列

a1 a2 a3 ...
b1 b2 b3 ...

に対してそれぞれの組み合わせの和を出力したいとします。
この時、次の実装で出力を試みましたが意図した出力になりませんでした。

a_list = map(int, input().split())
b_list = map(int, input().split())

res = 0

for a in list(a_list):
    print("a: {}".format(a))
    for b in list(b_list):
        print("b:{}".format(b))
        print("a:{} b:{} a+b:{}".format(a, b, a + b))

input

1 2
3 4

outout

a:1
b:3
a:1 b:3 a+b:4
b:4
a:1 b:4 a+b:5
a:2

expected

a:1
b:3
a:1 b:3 a+b:4
b:4
a:1 b:4 a+b:5
a:2
b:3
a:2 b:3 a+b:5
b:4
a:2 b:4 a+b:6

aのfor文の中でbのfor文が何度も繰り返し行われると思っていたのですが
1度しか現れない理由がわかりません。

どなたか教えていただければと思います。よろしくお願いします。

Answer 1

b_listはジェネレータなので、一度走りきると以降値を返しません。
外側のループの存在によって、list(b_list)が二回実行されていることが問題です。

>>> gen = map(int, ['1', '2', '3'])
>>> print(list(gen))
[1, 2, 3]
>>> print(list(gen))
[]

---

入力時にあらかじめリストとして受け取れば良いかと思います。

a_list = list(map(int, input().split()))
b_list = list(map(int, input().split()))

あるいは、リスト内包表記を使っても良いです。(個人的にはこちらが好み)

a_list = [int(e) for e in input().split()]
b_list = [int(e) for e in input().split()]

---

なお、要素の全ての組み合わせ(直積)を取る場合、itertools.productが便利です。

Comments

[shogiOtakku]

早い回答ありがとうございます！
なるほど、ジェネレータは一度しか値を返さないんですね。。。
ベストアンサーとさせていただきますm(__)m

[LouiS0616]

そう言えば。競プロっぽい問題ですが、出典はどちらでしょうか。
サイトによっては禁止されている場合もあるので、質問には前以て出典を書いておくと良いです。

[shogiOtakku]

出典を書くべきでしたね。。。すみません、次回から気を付けます。。。
AtCoderの過去問を解いている中で詰まったので質問しました。
https://atcoder.jp/tos
ただ今回は直接答えを聞くということでもなく、あくまでもpythonのお勉強という位置づけになります。

[LouiS0616]

AtCoderの過去問ならば問題なさそうですね。リンクもありがとうございます。
悲しいことにごく一部に規約違反をするユーザがいるために、回答側も少し神経質になっているきらいがあるのです。トラブルを未然に防ぐ意味でも出典を明記しておくと良いですね。

勉強頑張ってください。