python小技巧——enumerate,operator和Counter

最近看人家的代码，发现了很多python内置的函数和数据结构，可以大大减少我们的代码量（而且人家是优化过的）。python以简介和强大的库著称，要是什么都自己写的话，太不python了。这里介绍常用的能大大提高我们生活幸福度的小技巧：

enumerate()

这个函数主要用于既要遍历元素又要记下索引。通常我们都会这样写(不能再笨拙了)：

for i in range(0,len(list)):
    print i,list[i]

使用enumerate:

for index,text in enumerate(list):
    print index,text

itemgetter()

这个和下面的函数在operator库中，通常用在排序中。例如要对一个tuple的列表进行排序，找到第二个值最小的那个tuple。我一开始这样做：

list_of_tuples = [(1,2), (3,4), (5,0)]
min_tuple = None
minimum = sys.maxint
for pair in list_of_tuples:
    x,y = pair
    if y < minimum:
        min_tuple = pair
print min_tuple

后来，进化了，这样做：

def snd(pair):
    x,y = pair
    return y

list_of_tuples = [(1,2), (3,4), (5,0)]
min(list_of_tuples, key=snd)

这样看起来好多了，可是我发现人家的代码里是这样做的：

import operator
list_of_tuples = [(1,2), (3,4), (5,0)]
min(list_of_tuples, key=operator.itemgetter(1)) #use 2nd value

attgetter()

这个函数和上面的函数很像，不同的是，上面是用索引获取对应值，而这里使用属性来获取对应值，就像dict一样。
假如我们有个类Student:

class Student(object):
    def __init__(self, id, name, marks):
        self.id = id
            self.name = name
            self.marks = marks

    def __str__(self):
        return '%s has marks %s' %(self.name, self.marks)

我们有一个学生的实例的列表students，需要从里面找到分最高的。最简介的做法是:

students = [ Student(0, 'Foo', 30), Student(1, 'Bar', 95), Student(2, 'Baz', 80)]

best_student = max(students, key=operator.attrgetter('marks')) # don't forget the quotes

collections.Counter()

一看这个函数名，就知道是用来统计个数的~返回的是一个dict,key是各个元素，value是对应的个数。一开始，我并不知道这个函数，都是这样统计的：

d={}
for text in list:
    if not text in d.keys():
        d[text]=1
    else:
        d[text]+=1
print d

现在，直接一句话：

counter=collections.Counter(list)
print counter

之后获取各元素的个数，使用方法同一般的dict。如果增加元素e1,e2和对应个数，直接counter.update({e1=4,e2=3})就可以。

此外，这个Counter不仅用来数数，还有好多函数方便各种操作，例如:

• counter.most_common(n) 返回数目最多的前n个元素和对应个数
• a.substract(b) 返回一个Counter,Counter a减去Counter b，多的元素个数为正值，少的元素个数为负值
• counter.elements() 返回一个element列表，其中每个元素有多少个就重复多少次
• counter.values() 返回个数列表，通常配合sum(counter.values())
• counter.clear() 重置counter
• del counter[e]删除元素e和它的纪录
• 算数操作：+和- ，对应元素个数加减；|和&，两个counter的并集和交集