データ構造 中級

32. リスト内包表記

リスト内包表記とは

「for文を書いて、リストを1つずつappendしていく」という処理は非常によく登場するのですが、実はPythonにはそれをたった1行で書けてしまう専用の書き方があります。それがリスト内包表記です。[式 for 変数 in イテラブル]という形で書き、末尾にif条件を付ければ、特定の要素だけを絞り込むこともできます。最初は読みにくく感じるかもしれませんが、Pythonのコードでは頻出の書き方なので、少しずつ慣れていきましょう。

サンプルコード

squares = [x ** 2 for x in range(1, 6)]
print(squares)

evens = [x for x in range(10) if x % 2 == 0]
print(evens)
1つ目は1〜5それぞれを2乗した[1, 4, 9, 16, 25]というリストを作っています。2つ目はif条件を付けることで、0〜9の中から偶数だけを抽出したリストを作っています。

📝 練習問題

練習問題：リスト内包表記で1〜10のうち3の倍数だけのリストを作成してください。