当我们统计文本的词频时,通常需要查询当前词是否已经出现过,如果出现过,那么次数增1。通常我用dict来保存词和词频。我常用的方式是:

if word not in vocab_dict:
    vocab_dict[word] = 0
vocab_dict[word] += 1

用了很久觉得速度还行。 
后来看到有大神是这么用的:

if word not in vocab_dict.keys():
    vocab_dict[word] = 0
vocab_dict[word] += 1

感觉这样似乎合理一些,因为我要做的本来就是常看word是否是vocab_map的key.于是之后都用这种方式。 
某天,处理大量文本,且文本属于开放领域,词汇量也大,采用第二种方式,速度极慢,这时候,以为是查询dict本身比较慢,并没有意识到是vocab_map.keys()的问题。于是,将词同时存入set,查询时查set,速度快了很多。

if word not in vocab_set:
    vocab_dict[word] = 0
vocab_dict[word] += 1

抽取小部分数据,量化对比set、dict、dict.keys()的查询速度,结果如下: 
1.查询vocab_dict,打印语句:

print "time cost is %d ms." % ((end - begin).microseconds/1000)
time cost is 175 ms.

2.查询vocab_dict.keys(),打印语句如下:

print "time cost is %d s." % ((end - begin).seconds)
time cost is 45 s.

3.查询vocab_set,打印语句:

print "time cost is %d ms." % ((end - begin).microseconds/1000)
time cost is 168 ms.

和查询vocab_dict差不多。

那么为什么dict.keys()查询速度比另外两个慢很多呢?这就要对比list、dict、set三种的数据结构了。 
dict.keys()实际上是list(keys),是dict的所有key组成的list。查找一个元素是否在list中是以list的下标为索引遍历list.而查询是否在dict中,是将key以hash值的形式直接找到key对应的索引,根据索引可直接访问value。对量大的dict查询,自然是后者快很多。 
而set和dict的存储原理基本是一样的,唯一不同的是,set没有value,只有key。对查询key是否在dict或sset内,效果基本上是一样的。

由此,可以得出,如果存储的数据会被反复查询,且量大,那么,尽量不要用list,尽量用dict,如果元素不重复,用set更好。查询是否在dict内,也不要在用dict.keys()