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

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

ただいまの
回答率

87.80%

【python3】内包表記でリストを連結させたい

解決済

回答 3

投稿 編集

  • 評価
  • クリップ 0
  • VIEW 243

score 1256

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

def hoge(suuji):
    if suuji == 1:
        return [1, 2]
    if suuji == 2:
        return [3, 4]
    if suuji == 3:
        return [5, 6]
    return [7, 8]

print([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

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

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 3

+2

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


で。

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

checkベストアンサー

+1

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

import itertools

def hoge(suuji):
    if suuji == 1:
        return [1, 2]
    if suuji == 2:
        return [3, 4]
    if suuji == 3:
        return [5, 6]
    return [7, 8]


l = []
for i in range(1,5):
    l.extend(hoge(i))
print(l) # [1, 2, 3, 4, 5, 6, 7, 8]

l = list(itertools.chain.from_iterable([hoge(i) for i in range(1,5)]))
print(l) # [1, 2, 3, 4, 5, 6, 7, 8]

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

nested_list = [[1,2],[3]]

# [e for inner_list in nested_list for e in inner_list] は 以下と同じ
l = []
for inner_list in nested_list:
    print(inner_list)
    for e in inner_list:
        l.append(e)

print(l) # [1,2,3]

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2020/01/21 16:04

    ありがとうございます。

    extendsはすぐに思いついたのですが、
    内包表記化した際にはどうすればいいのか気になって質問をしてみました。

    また、チームでの決めがあり、勝手にimportができない状況なのです。。。

    正確に情報をお伝えせずに質問をしてしまい申し訳ございません。

    キャンセル

  • 2020/01/21 16:17

    回答に追記しました。通常のfor文に書き換えると分かりやすいかと思います。

    キャンセル

+1

最初の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]

こんな感じでしょうか?

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

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

  • ただいまの回答率 87.80%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る