Pythonでリストのリストを一行で平坦化する方法

PythonでPythonを扱う際、よくある状況があります。

リストのリストがあり、それを単一のリストにフラット化したい場合です。

問題

例を示します。

data = [[1, 2], [3, 4], [5, 6]]

期待される出力：

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

リストのリストをフラット化する最もPython的な方法は次のとおりです。

flattened = [item for sublist in data for item in sublist]

これは入れ子になったリスト内包表記であり、機能的には次のものと等価です。

flattened = []

for sublist in data:

    for item in sublist:

        flattened.append(item)

しかし、はるかに簡潔な形式です。

より大きく、メモリを意識する必要があるデータセットの場合、Pythonの組み込みitertoolsモジュールを使用できます。

import itertools

flattened = list(itertools.chain.from_iterable(data))

このアプローチはイテレータを作成するため、大量のデータ処理を行う際に効率的です。

data = [['a', 'b'], ['c', 'd'], ['e']]

flattened = [item for sublist in data for item in sublist]

print(flattened)

# 出力: ['a', 'b', 'c', 'd', 'e']

このテクニックはクリーンなPython開発の定番であり、データ操作、Webスクレイピング、データパイプラインなどで広く使用されています。