Pythonでタプルのリスト内のタプル出現回数をカウントする方法

Pythonでは、リストの要素としてタプルを持つデータ構造を扱うことがあります。本記事では、そのようなリストの中に存在する一意なタプルの種類ごとに出現回数をカウントする方法を、2つのアプローチから解説します。

defaultdictを使う方法

collectionsモジュールのdefaultdictは、キーが存在しない場合にデフォルト値を自動的に設定してくれる便利なコンテナです。与えられたリストを走査しながら、各タプルをキーとして出現回数を累積していきます。

サンプルコード

import collections
Alist = [[('Mon', 'Wed')], [('Mon')], [('Tue')],[('Mon', 'Wed')] ]
# 対象のリスト
print("Given list:\n", Alist)
res = collections.defaultdict(int)
for elem in Alist:
    res[elem[0]] += 1
print("Count of tuples present in the list:\n",res)

実行結果

上記のコードを実行すると、以下のような結果が得られます。

Given list:
[[('Mon', 'Wed')], ['Mon'], ['Tue'], [('Mon', 'Wed')]]
Count of tuples present in the list:
defaultdict(, {('Mon', 'Wed'): 2, 'Mon': 1, 'Tue': 1})

この方法では、初期値0が自動的に設定されるため、キーの存在チェックを明示的に行う必要がなく、シンプルなコードでカウント処理を実現できます。

Counterとchainを組み合わせる方法

もうひとつの方法は、collectionsモジュールのCounterとitertoolsモジュールのchainを組み合わせる手法です。chain(*Alist)によってネストされたリストを平坦化し、その結果をCounterに渡すことで、各タプル要素の出現回数を一度に取得できます。

サンプルコード

from collections import Counter
from itertools import chain
Alist = [[('Mon', 'Wed')], [('Mon')], [('Tue')],[('Mon', 'Wed')] ]
# 対象のリスト
print("Given list:\n", Alist)
res = Counter(chain(*Alist))
print("Count of tuples present in the list:\n",res)

実行結果

上記のコードを実行すると、以下のような結果が得られます。

Given list:
[[('Mon', 'Wed')], ['Mon'], ['Tue'], [('Mon', 'Wed')]]
Count of tuples present in the list:
Counter({('Mon', 'Wed'): 2, 'Mon': 1, 'Tue': 1})

まとめ

どちらの方法でも同じ結果が得られますが、Counterとchainを組み合わせる方法は1行で書けるため簡潔です。一方、defaultdictを使う方法は処理の流れが明示的で、カスタマイズしやすいという利点があります。用途やコードの可読性の要件に応じて使い分けるとよいでしょう。