« ^ »

Pyhtonのcollections.Counterで数を数える

所要時間: 約 1分

collections.Counterで数を数える

ログなどを集計していくと出現個数を数えたい場合がある。 例えば以下のようなデータのstatusの数を数えたいとする。

data = [
    {'tag': 'foo', 'second': 1, 'status': 'success'},
    {'tag': 'bar', 'second': 1, 'status': 'success'},
    {'tag': 'baz', 'second': 4, 'status': 'success'},
    {'tag': 'foo', 'second': 1, 'status': 'success'},
    {'tag': 'bar', 'second': 1, 'status': 'success'},
    {'tag': 'baz', 'second': 4, 'status': 'success'},
    {'tag': 'foo', 'second': 1, 'status': 'fail'},
    {'tag': 'bar', 'second': 1, 'status': 'success'},
    {'tag': 'baz', 'second': 4, 'status': 'success'},
    {'tag': 'foo', 'second': 1, 'status': 'fail'},
    {'tag': 'bar', 'second': 1, 'status': 'success'},
    {'tag': 'baz', 'second': 4, 'status': 'success'},
    {'tag': 'foo', 'second': 1, 'status': 'fail'},
    {'tag': 'bar', 'second': 1, 'status': 'success'},
    {'tag': 'baz', 'second': 4, 'status': 'success'},
]

for文でその要素が出てきたらカウンタをインクリメントするコードは次のようになる。

>>> counter = {}
>>> for record in data:
...    status = record.get('status', 'unknown')
...    if status not in counter:
...        counter[status] = 0
...    counter[status] += 1
...
>>> print(counter)
{'success': 12, 'fail': 3}

collections.Counter()を使う

collections.Counter()を使うと上記のコードを完結に書ける。

>>> import collections
>>>
>>> counter = collections.Counter()
>>> for record in data:
...    counter.update([record.get('status', 'unknown')])
...
>>> print(counter)
Counter({'success': 12, 'fail': 3})

初期化時に要素を渡すことが可能であれば、さらに完結に記述できる。

>>> import collections
>>>
>>> counter = collections.Counter(record.get('status', 'unknown') for record in data)
>>> print(counter)
Counter({'success': 12, 'fail': 3})