😊

【初心者向け】Pythonのリスト内包表記について説明

2024/10/17に公開

Pythonのリスト内包表記(リストコンプリヘンション)は、リストをシンプルかつ効率的に生成するための強力な機能です。コードを簡潔にし、処理速度も向上させることができるため、Pythonプログラマーにとって習得しておきたいテクニックの一つです。本記事では、リスト内包表記の基本から応用的な使い方までを解説します。

リスト内包表記とは?

リスト内包表記は、リストを生成するための簡潔な方法です。通常の for ループや map 関数を使ってリストを生成する代わりに、一行で新しいリストを作成することができます。
基本構文は以下の通りです。

for 変数 in イテラブル(リストとか)

サンプル

# 普通のforループで代入
squares=[]
for i in range(10):
    squares.append(i ** 2)
print(squares) # 出力[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

# リスト内包表記
squares_comp = [x ** 2 for x in range(10)]
print(squares_comp)  #出力[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

同じ結果を得られますが、リスト内包表記を使うことでコードが非常に簡潔になります。

条件付きリスト内包表記

リスト内包表記は、条件を加えることもできます。for ループに条件を付けたい場合は、if 文をリスト内包表記の中に含めることが可能です。

構文:

for 変数 in イテラブル if 条件

サンプル:

# 偶数のみを出力
even_numbers = [x for x in range(10) if x % 2 == 0]
print(even_numbers) # 出力[0, 2, 4, 6, 8]

このように、条件付きでリストを生成する場合にもリスト内包表記は非常に便利です。

ネストされたリスト内包表記

リスト内包表記は、ネストされたループにも対応しています。二重ループやそれ以上のループを一行で書くことが可能です。

構文:

for 変数1 in イテラブル1 for 変数2 in イテラブル2

サンプル:

matrix = [[1, 2, 3],[4, 5, 6],[7, 8, 9]]
# 普通の二重ループでリストを作成
flatten_list = []
for row in matrix:
    for x in row:
        flatten_list.append(x)
print(flatten_list) # 出力[1, 2, 3, 4, 5, 6, 7, 8, 9]

# リスト内包表記の二重ループでリストを作成
flatten = [x for row in matrix for x in row]
print(flatten) # 出力[1, 2, 3, 4, 5, 6, 7, 8, 9]

この例では、2次元のリスト(行列)を1次元にフラット化しています。二重の for ループを使って処理していますが、リスト内包表記により非常に簡潔に記述できます。

応用例:条件付きネストされたリスト内包表記

ネストされたリスト内包表記でも条件を追加することが可能です。たとえば、特定の条件を満たす要素だけを抽出することができます。

matrix = [[1, 2, 3],[4, 5, 6],[7, 8, 9]]
odd_numbers = [x for row in matrix for x in row if x % 2 == 1]
print(odd_numbers) # 出力[1, 3, 5, 7, 9]

このように、複雑な条件付きのリスト生成もリスト内包表記で簡単に表現できます。

リスト内包表記とパフォーマンス

リスト内包表記は、for ループよりも一般的に高速です。特にリスト生成を大量に行う場合、リスト内包表記を使うことでパフォーマンスの向上が期待できます。

例:速度比較

import time
start = time.time()
squares = []
for x in range(1000000):
    squares.append(x ** 2)
end = time.time()
print(f'通常のループ: {end - start:.5f} 秒') # 出力通常のループ: 0.46409 秒

start = time.time()
squares_comp = [x ** 2 for x in range(1000000)]
end = time.time()
print(f'リスト内包ループ: {end - start:.5f} 秒') #リスト内包ループ: 0.33020 秒

この例を実行すると、通常の for ループよりもリスト内包表記の方が高速であることが確認できます。特に大規模なデータ処理において、パフォーマンス改善に役立ちます。

他の内包表記との比較

Pythonにはリスト内包表記以外にも、辞書やセット、ジェネレータの内包表記があります。それぞれ、リスト内包表記と同じように簡潔に書けるため、場面に応じて使い分けると良いでしょう。
セット:順序が不定で、重複する要素を持たないデータ構造です。数学の集合と似た概念で、要素の有無を素早く調べたり、重複を取り除いたりする際に便利です。
ジェネレータ:必要な時に要素を一つずつ生成するイテレータの一種です。リストのように一度にすべての要素をメモリに読み込むのではなく、必要に応じて要素を生成するため、メモリ効率が良いです。

辞書内包表記

# 辞書内包表記
squared_dict = {x: x ** 2 for x in range(5)}
print(squared_dict)  # 出力: {0: 0, 1: 1, 2: 4, 3: 9, 4: 16}

セット内包表記

# セット内包表記
squared_set = {x ** 2 for x in range(5)}
print(squared_set)  # 出力: {0, 1, 4, 9, 16}

ジェネレータ内包表記

# ジェネレータ内包表記(リストではなくイテレータを返す)
squared_gen = (x ** 2 for x in range(5))
print(list(squared_gen))  # 出力: [0, 1, 4, 9, 16]

リスト内包表記の注意点

リスト内包表記は強力ですが、適切に使わないとコードの可読性が下がることがあります。特に、複雑なロジックを含む場合は、通常の for ループを使ってコードを明示的に書く方が良い場合もあります。

ネストが深すぎる場合:ネストされたループや複雑な条件が多いと、内包表記が読みにくくなります。
過度な使用は避ける:簡潔さを追求しすぎると、逆にコードが理解しにくくなることがあります。

まとめ

Pythonのリスト内包表記は、シンプルで効率的なリスト生成を可能にします。単純なリストの作成から、条件付きやネストされたループまで、リスト内包表記を使うことでコードをより読みやすく、かつ高速に実行できるようになります。ただし、可読性を保つために、あまりにも複雑な場合には通常の for ループを使うなど、適切なバランスを保つことが大切です。

このテクニックをマスターすることで、Pythonのコードをさらに洗練されたものにしていきましょう!

Discussion