Pythonでネストされたリストの全要素数をカウントする方法

データ処理を行う際、リストに含まれる要素の数を数える必要がある場面は多くあります。しかし、リストがネスト（入れ子）構造になっている場合、単純に len() 関数を呼び出すだけでは全体の要素数を正しく取得できません。本記事では、こうした複雑なケースにも対応できる、リスト内の全要素数をカウントする方法を2つ紹介します。

forループ（リスト内包表記）を使う方法

この方法では、リスト内包表記を使ってネスト構造を走査し、リストを平坦化（フラット化）します。以下のプログラムでは、内側の各リストが異なる数の要素を持つネストされたリストを扱っています。平坦化したリストに対して len() 関数を適用することで、全体の要素数を求めることができます。

サンプルコード

listA = [[2,9, 6], [5, 'a'], [0], [12,'c', 9, 3]]

# 対象のリスト
print("Given list : ",listA)

res = len([x for y in listA for x in y])

# 結果を出力
print("Total count of elements : " ,res)

実行結果

上記のコードを実行すると、次のような出力が得られます。

Given list : [[2, 9, 6], [5, 'a'], [0], [12, 'c', 9, 3]]
Total count of elements : 10

itertools.chain を使う方法

この方法では、標準ライブラリ itertools の chain メソッドを使用します。chain(*listA) を呼び出すことで、ネストされたリストからすべての内側の要素を取り出して平坦化し、それを list() 関数でリストに変換します。最後に len() 関数を適用すれば、リスト内の全要素数を取得できます。

サンプルコード

from itertools import chain
listA = [[2,9, 6], [5, 'a'], [0], [12,'c', 9, 3]]

# 対象のリスト
print("Given list : ",listA)

res = len(list(chain(*listA)))

# 結果を出力
print("Total count of elements : " ,res)

実行結果

上記のコードを実行すると、次のような出力が得られます。

Given list : [[2, 9, 6], [5, 'a'], [0], [12, 'c', 9, 3]]
Total count of elements : 10

まとめ

どちらの方法でも同じ結果が得られます。リスト内包表記は追加のインポートが不要で手軽に使えるのが利点です。一方、itertools.chain を使う方法は大きなデータセットでもメモリ効率が良いという特徴があります。なお、内側の要素がすべてリストであることが保証されている場合は、sum(len(sub) for sub in listA) のように各サブリストの長さを合計する方法も利用できます。用途やデータの規模に応じて、最適な方法を選択してください。