Python语言中计数方法的演变

有时候，利用Python语言简洁、优雅地解决问题的方法，会随着时间变化。随着Python不断进化，统计列表元素数量的方法也在改变。

以计算元素在列表中出现的次数为例，我们可以编写出许多不同的实现方法。在分析这些方法时，我们先不关注性能，只考虑代码风格。

要理解这些不同的实现方式，我们得先知道一些历史背景。幸运的是，我们生活在" future "世界，拥有一台时间机器。接下来，我们一起坐上时光机，回到1997年吧。

if 语句

1997年1月1日，我们使用的是Python 1.4。现在有一个不同颜色组成的列表，我们想知道列表里每种颜色出现的次数。我们用 字典 来计算吧！

colors = ["brown", "red", "green", "yellow", "yellow", "brown", "brown", "black"] color_counts = {} for c in colors:     if color_counts.has_key(c):         color_counts[c] = color_counts[c] + 1     else:         color_counts[c] = 1

注意：我们没有使用 += ，因为增量赋值直到Python 2.0才出现；另外，我们也没有使用 c in color_counts 这个惯用法（idiom），因为这也是Python 2.2中才发明的，

运行上述代码之后，我们会发现 color_counts 字典里，现在包含了列表中每种颜色的出现次数。

>>> color_counts {'brown': 3, 'yellow': 2, 'green': 1, 'black': 1, 'red': 1}

上面的实现很简单。我们遍历了每一种颜色，并判断该颜色是否在字典中。如果不在，就在字典加入该颜色；如果在，就增加这种颜色的计数。

我们还可以把上面的代码改写为：

colors = ["brown", "red", "green", "yellow", "yellow", "brown", "brown", "black"] color_counts = {} for c in colors:     if not color_counts.has_key(c):         color_counts[c] = 0     color_counts[c] = color_counts[c] + 1

如果列表稀疏度高（即列表中不重复的颜色数量很多），这段代码可能运行的会有点慢。因为我们现在要执行两个语句，而不是一个。但是我们不关心性能问题，我们只关注编码风格。经过思考，我们决定采用新版的代码。

try 代码块（Code Block）

1997年1月2日，我们使用的还是Python 1.4。今早醒来的时候，我们突然意识到：我们的代码遵循的是“三思而后行”（Look Before You Leap，即事先检查每一种可能出现的情况）原则，但实际上我们应该按照“获得谅解比获得许可容易”（Easier to Ask Forgiveness, Than Permission，即不检查，出了问题由异常处理来处理）的原则进行编程，因为后者更加简洁、优雅。我们用 try-except 代码块来重构下代码吧：

colors = ["brown", "red", "green", "yellow", "yellow", "brown", "brown", "black"] color_counts = {} for c in colors:     try:         color_counts[c] = color_counts[c] + 1     except KeyError:         color_counts[c] = 1

现在，我们的代码尝试增加每种颜色的计数。如果某颜色不在字典里，那么就会抛出KeyError，我们随之将该颜色的计数设置为1。

get 方法

1998年1月1日，我们已经升级到了Python 1.5。我们决定重构之前的代码，使用字典中新增的 get 方法。

colors = ["brown", "red", "green", "yellow", "yellow", "brown", "brown", "black"] color_counts = {} for c in colors:     color_counts[c] = color_counts.get(c, 0) + 1

现在，我们的代码会遍历每种颜色，从字典中获取该颜色的当前计数值。如果没有这个计数值，则该颜色的计数值默认为0，然后在数值的基础上加1。最后将字典中相应键的值设置为新的计数。

把主要代码都写在一行里，感觉很酷，但是我们不敢完全肯定这种做法更加简洁、优雅。我们觉得可能有点太聪敏了，所以还是撤销了这次的重构。

setdefault 方法

2001年1月1日，我们现在使用的是Python 2.0。我们听说字典类型现在有一个 setdefault 方法，决定利用它重构我们的代码。我们还决定使用新增加的 += 增量赋值运算符。

colors = ["brown", "red", "green", "yellow", "yellow", "brown", "brown", "black"] color_counts = {} for c in colors:     color_counts.setdefault(c, 0)     color_counts[c] += 1

无论是否需要，我们在每一次循环时都会调用 setdefault 方法。但这样做，的确会让代码看上去可读性更高。我们发现这种方法比之前的代码更加简洁、优雅，所以提交了此次修改。

fromkeys 方法

2004年1月1日，我们使用的是Python 2.3。我们听说字典新增了一个叫 fromkeys 的类方法（class method），可以利用列表中的元素作为键来构建字典。我们使用新方法重构了代码：

colors = ["brown", "red", "green", "yellow", "yellow", "brown", "brown", "black"] color_counts = dict.fromkeys(colors, 0) for c in colors:     color_counts[c] += 1

这段代码将不同的颜色作为键，创建了一个新的字典，每个键的值被默认设置为0。这样，我们增加每个键的值时，就不用担心是否已经进行了设置。我们也不需要在代码中进行检查或异常处理了，这看上去是个改进。我们决定就这样修改代码。

推导式（Comprehension）与集合

2005年1月1日，我们现在用的是Python 2.4。我们发现可以利用集合（Python 2.3中发布，2.4版成为内置类型）与列表推导式（Python 2.0中发布）来解决计数问题。进一步思考之后，我想起来Python 2.4中还发布了生成器表达式（generator expressions），我们最后决定不用列表推导式，而是采用生成器表达式。

colors = ["brown", "red", "green", "yellow", "yellow", "brown", "brown", "black"] color_counts = dict((c, colors.count(c)) for c in set(colors))

注意：我们这里使用的不是字典推导式，因为字典推导式直到Python 2.7才被发明。

运行成功了，而且只有一行代码。但是这种代码够简洁、优雅吗 ？

我们想起了Python之禅（Zen of Python），这个Python编程指导原则起源于一个Python邮件列表，并悄悄地收进了Python 2.2.1版本中。我们在REPL（read-eval-print loop，交互式解释器）界面中输入 import this ：

>>> import this The Zen of Python, by Tim Peters  Beautiful is better than ugly. # 优美胜过丑陋 Explicit is better than implicit. # 明确胜过含蓄 Simple is better than complex. # 简单胜过复杂 Complex is better than complicated. # 复杂胜过难懂 Flat is better than nested. # 扁平胜过嵌套 Sparse is better than dense. # 稀疏胜过密集 Readability counts. # 易读亦有价 Special cases aren't special enough to break the rules. # 特例也不能特殊到打破规则 Although practicality beats purity. # 尽管实用会击败纯洁 Errors should never pass silently. # 错误永远不应默默地溜掉  Unless explicitly silenced. # 除非明确地使其沉默 In the face of ambiguity, refuse the temptation to guess.  # 面对着不确定，要拒绝猜测的诱惑 There should be one-- and preferably only one --obvious way to do it.  # 应该有一个–宁肯只有一个–明显的实现方法 Although that way may not be obvious at first unless you're Dutch. # 也许这个方法开始不是很明显，除非你是荷兰人 Now is better than never. #现在做也要胜过不去做 Although never is often better than right now. # 尽管不做通常好过立刻做 If the implementation is hard to explain, it's a bad idea. # 如果实现很难解释，那它就是一个坏想法 If the implementation is easy to explain, it may be a good idea. # 如果实现容易解释，那它可能就是一个好想法 Namespaces are one honking great idea -- let's do more of those! # 命名空间是一个响亮的出色想法–就让我们多用用它们

我们的代码变得更复杂，时间复杂度从O(n)增加到了O(n ² )；还变的更丑，可读性更差了。我们那样改，是一次有趣的尝试，但是一行代码的实现形式，没有我们之前的方法简洁、优雅。我们最后还是决定撤销修改。

defaultdict 方法

2007年1月1日，我们使用的是Python 2.5。我们刚发现， defaultdict 已经被加入标准库。这样，我们就可以把字典的默认键值设置为0了。让我们使用 defaultdict 重构代码：

from collections import defaultdict     colors = ["brown", "red", "green", "yellow", "yellow", "brown", "brown", "black"] color_counts = defaultdict(int) for c in colors:         color_counts[c] += 1

那个 for 循环现在变得真简单！这样肯定是更加简洁、优雅了。

我们发现， color_counts 这个变量的行为现在有点不同，但是它的确继承了字典的特性，支持所有相同的映射功能。

>>> color_counts defaultdict(<type 'int'>, {'brown': 3, 'yellow': 2, 'green': 1, 'black': 1, 'red': 1})

我们在这里没有把 color_counts 转换成字典，而是假设其他的代码也使用鸭子类型（duck typing， Python中动态类型的一种，这里的意思是：其他代码会将 color_counts 视作字典类型），不再改动这个类似字典的对象。

Counter 类

2011年1月1日，我们使用的是Python 2.7。别人告诉我们，之前使用 defaultdict 编写的代码，不再是统计颜色出现次数最简洁、优雅的方法了。Python 2.7中新引入了一个 Counter 类，可以完全解决我们的问题。

from collections import Counter colors = ["brown", "red", "green", "yellow", "yellow", "brown", "brown", "black"] color_counts = Counter(colors)

还有比这更简单的方法吗？这个一定是最简洁、优雅的实现了。

与 defaultdict 一样， Counter 类返回的也是一个类似字典的对象（实际是字典的一个子类）。这对满足我们的需求来说足够了，所以我们就这么干了。

>>> color_counts Counter({'brown': 3, 'yellow': 2, 'green': 1, 'black': 1, 'red': 1})

性能考虑

请注意，在编写这些实现方式时，我们都没有关注效率问题。大部分方法的时间复杂度相同（O(n)），但是不同的Python语言实现形式（如CPython, PyPy，或者Jython）下，运行时间会有差异。

尽管性能不是我们的主要关注点，我还是在CPython 3.5.0的实现下测试了 运行时间 。从中，你会发现一个有趣的现象：随着列表中颜色元素的密度（即相同元素的数量）变化，每一种实现方法的相对效率也会不同。

结语

根据Python之禅，“ 应该有一个——宁肯只有一个明显的实现方法”。这句话所说的状态值得追求，但事实是，并不总是只有一种明显的方法。这个“明显”的方法会随着时间、需求和专业水平，不断地变化。

“简洁、优雅”（即Pythonic）也是一个相对的概念。