Question

假设我有一个全部共享密钥color和object的词典列表：

inpt = [{'object':'square', 'color':'red', 'size':'big'},
        {'object':'square', 'color':'red', 'coord':(0,0)},
        {'object':'square', 'color':'red'},
        {'object':'triangle', 'color':'blue', 'adj':'beautiful'},
        {'object':'triangle', 'color':'blue', 'attr':'none'}]

此处我只关心object和color。我想计算一下红色方块和蓝色三角形的数量。这意味着查找{'object':'square', 'color':'red'}和{'object':'triangle', 'color':'blue'}发生的次数。换句话说，我需要在像这样的字典列表中找到键值对中的共性数。

最终结果可能是这样的：

{('square', 'red'): 3, ('triangle', 'blue'):2}

这是一种简单的方法吗？

Answer 1

只需使用collections.Counter() object，为其提供值的元组：

from collections import Counter

result = Counter((d['object'], d['color']) for d in inpt)

这为您提供了一个带有(object, color)键的字典子类。您可以使用Counter.most_common() method：

按递减计数顺序获取元组列表

result = result.most_common()

演示：

>>> from collections import Counter
>>> inpt = [{'object':'square', 'color':'red', 'size':'big'},
...         {'object':'square', 'color':'red', 'coord':(0,0)},
...         {'object':'square', 'color':'red'},
...         {'object':'triangle', 'color':'blue', 'adj':'beautiful'},
...         {'object':'triangle', 'color':'blue', 'attr':'none'}]
>>> Counter((d['object'], d['color']) for d in inpt)
Counter({('square', 'red'): 3, ('triangle', 'blue'): 2})
>>> _.most_common()
[(('square', 'red'), 3), (('triangle', 'blue'), 2)]

计算字典列表中的公共键值对

1 个答案: