DS JUNKSTAブログ

元文系院卒が独学で取り組むDS(データサイエンス)、その有象無象のアウトプット、DSは本当に楽しい

【Python】リスト内包表記(list comprehension)

Pythonでは「リスト内包表記」を用いたら、
リストの中でfor文でイテラブルオブジェクトを回して新しいリストを生成できる。
「リスト内包表記」は英語で list comprehension 。

for文を用いたリスト内包表記の基本形
[処理 for 変数 in イテラブルオブジェクト]

例えば、

squares = []
for x in range(10):
    squares.append(x**2)
# [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

のように複数行で書かれるfor文のコードは、

squares = [x**2 for x in range(10)]
# [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

の1行でかける。

参考にさせていただいたWEB記事

Pythonリスト内包表記の使い方 | note.nkmk.me