欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python下defaultdict方法使用介绍

程序员文章站 2022-09-15 23:04:32
本文主要参考最下面的网页,简单做个提炼。 defaultdict主要解决在创建的dict为空时的赋值问题。 (1)dict.setdefault()方法 当不用default...

本文主要参考最下面的网页,简单做个提炼。

defaultdict主要解决在创建的dict为空时的赋值问题。

(1)dict.setdefault()方法

当不用defaultdict方法时,可用dict.setdefault()方法替代

下面使用参考资料中的例子举例。例子是用来统计数组中每个词出现的次数。setdefault(kw,0)方法可在dict中没有kw这个key的时候,将这个key的value设置为0,并且当dict中存在kw时,则返回其对应的value。

strings = ('puppy', 'kitten', 'puppy', 'puppy', 'weasel', 'puppy', 'kitten', 'puppy')
counts = {}
for kw in strings:
    counts[kw] = counts.setdefault(kw, 0) + 1

(2)defaultdict方法

defaultdict()方法可以接受一个类型作为参数,比如int, list等,也可以接受一个不带参数的函数作为参数。

举例1:以int作为参数,统计单词出现次数

from collections import defaultdict

strings = ('puppy', 'kitten', 'puppy', 'puppy',
           'weasel', 'puppy', 'kitten', 'puppy')
counts = defaultdict(int)  

for s in strings:
    counts[s] += 1

print(counts) # 输出是defaultdict(, {'puppy': 5, 'kitten': 2, 'weasel': 1})

举例2:以一个无参数函数作为参数,同样是统计单词出现的次数

from collections import defaultdict

strings = ('puppy', 'kitten', 'puppy', 'puppy',
           'weasel', 'puppy', 'kitten', 'puppy')

def zero():
    return 0

counts = defaultdict(zero)  # zero是函数名

for s in strings:
    counts[s] += 1

print(counts) # 输出是defaultdict(, {'puppy': 5, 'kitten': 2, 'weasel': 1})