初歩的な質問で恐縮です。

python
1def hoge(suuji):
2    if suuji == 1:
3        return [1, 2]
4    if suuji == 2:
5        return [3, 4]
6    if suuji == 3:
7        return [5, 6]
8    return [7, 8]
9
10print([hoge(suuji) for suuji in {1, 2, 3, 4}])

上記のプログラムを実行すると[[1, 2], [3, 4], [5, 6], [7, 8]]と出力されました。

hoge関数の中身を変えられない状況なのですが、
内包表記側をどうにかいじって[1, 2, 3, 4, 5, 6, 7, 8]と出力させるには
どうすればいいでしょうか。

連結される順番はどうであっても構いませんが、1つのリストで欲しいです。

追記

このサイトを見てできましたが、なぜ動くのかが分かりません。

どなたかご解説いただけないでしょうか。

さらに追記

ぐぐり方が悪かったみたいです。
ようやく理解できました。表記を知らなかっただけでした。
https://qiita.com/propella/items/fa64b40b6f45d4f32cbc

ご回答皆様、誠にありがとうございました。

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

回答3件

python
1print([x for suuji in {1, 2, 3, 4} for x in hoge(suuji)])

で。

投稿2020/01/21 07:19

quickquip

総合スコア11299

最初のforで

[1, 2]
[3, 4]
[5, 6]
[7, 8]

を取り出し

次のforで
[e for e in [1 ,2]] + [e for e in [3 ,4]] + [e for e in [5 ,6]] + [e for e in [7 ,8]]

[1, 2] + [3, 4] + [5, 6] + [7, 8]

[1, 2, 3, 4, 5, 6, 7, 8]

こんな感じでしょうか？

投稿2020/01/21 08:21

barobaro

総合スコア1286

ベストアンサー

素直にextendを使ったり便利なitertools.chain.from_iterableを使ったりするとできます。

Python
1import itertools
2
3def hoge(suuji):
4    if suuji == 1:
5        return [1, 2]
6    if suuji == 2:
7        return [3, 4]
8    if suuji == 3:
9        return [5, 6]
10    return [7, 8]
11
12
13l = []
14for i in range(1,5):
15    l.extend(hoge(i))
16print(l) # [1, 2, 3, 4, 5, 6, 7, 8]
17
18l = list(itertools.chain.from_iterable([hoge(i) for i in range(1,5)]))
19print(l) # [1, 2, 3, 4, 5, 6, 7, 8]

なお、二重の内包表記を使った方法の理解は、以下のようにfor文に展開すると理解しやすいかと思います。

Python
1nested_list = [[1,2],[3]]
2
3# [e for inner_list in nested_list for e in inner_list] は 以下と同じ
4l = []
5for inner_list in nested_list:
6    print(inner_list)
7    for e in inner_list:
8        l.append(e)
9
10print(l) # [1,2,3]