本文实例讲述了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不仅用来数数,还有好多函数方便各种操作,例如:
希望本文所述对大家Python程序设计有所帮助。
Powered By python教程网 鲁ICP备18013710号
python博客 - 小白学python最友好的网站!