collections.Counter,有什么办法可以避免添加字符串值?

时间:2020-02-16 20:37:06

标签: python dictionary collections counter

例如,我有两个字典。

>>> dict_a = {'total': 20, 'remaining': 10, 'group': 'group_a'}
>>> dict_b = {'total': 30, 'remaining': 29, 'group': 'group_a'}

我使用collections.Counter进行计数。

>>> dict_c = Counter()
>>> dict_c.update(Counter(dict_a))
>>> dict_c.update(Counter(dict_b))
>>> print(dict_c)
{'toal': 50, 'remaining': 39, 'group': 'group_agroup_a'}

有什么办法可以只添加整数类型的值?即添加时仅会添加整数类型值。

>>> print(dict_c)
>>> {'toal': 50, 'remaining': 39, 'group': 'group_a'}

2 个答案:

答案 0 :(得分:1)

您可以定义自己的函数,以添加两个Counter对象,例如问题中的对象。这是必要的,因为添加Counter对象的默认方法无法像您放入对象一样处理其中的非数字值。

from collections import Counter

def add_counters(a, b):
    """ Add numerical counts from two Counters. """
    if not isinstance(a, Counter) or not isinstance(a, Counter):
        return NotImplemented
    result = Counter()
    for elem, count in a.items():
        newcount = count + b[elem]
        try:
            if newcount > 0:
                result[elem] = newcount
        except TypeError:
            result[elem] = count  # Just copy value.

    for elem, count in b.items():
        if elem not in a and count > 0:
            result[elem] = count

    return result


dict_a = {'total': 20, 'remaining': 10, 'group': 'group_a'}
dict_b = {'total': 30, 'remaining': 29, 'group': 'group_a'}
dict_c = add_counters(Counter(dict_a), Counter(dict_b))
print(dict_c)  # -> Counter({'total': 50, 'remaining': 39, 'group': 'group_a'})

请注意,上述内容可能并不完全正确,因为第一个Counter参数a中任何刚刚复制到结果中的非数字项都可能被第二个for覆盖循环,使它们的最终值等于名为Counter的第二个b中的值。之所以这样,是因为您没有精确定义在这种情况下想要发生的事情。

答案 1 :(得分:0)

有什么办法可以只添加整数类型的值?

这可能不是最有效的解决方案,但是您可以简单地遍历dict_c的键值对来检查值是否为int类型,以创建仅包含以下内容的新字典整数值。

from collections import Counter

dict_a = {'total': 20, 'remaining': 10, 'group': 'group_a'}
dict_b = {'total': 30, 'remaining': 29, 'group': 'group_a'}
dict_c = Counter(dict_a) + Counter(dict_b)
dict_result = {key: value for key, value in dict_c.items() if isinstance(value, int)}
print(dict_result)

这将返回预期结果:

{'total': 50, 'remaining': 39}