本文實例講述了python函數(shù)enumerate,operator和Counter使用技巧。分享給大家供大家參考，具體如下：

最近看人家的代碼，發(fā)現(xiàn)了很多python內(nèi)置的函數(shù)和數(shù)據(jù)結(jié)構(gòu)，可以大大減少我們的代碼量（而且人家是優(yōu)化過的）。python以簡潔和強大的庫著稱，要是什么都自己寫的話，太不python了。這里介紹常用的能大大提高我們生活幸福度的小技巧：

enumerate()

這個函數(shù)主要用于既要遍歷元素又要記下索引。通常我們都會這樣寫(不能再笨拙了)：

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

使用enumerate:

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

itemgetter()

這個和下面的函數(shù)在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)

這樣看起來好多了，可是我發(fā)現(xiàn)人家的代碼里是這樣做的：

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

attgetter()

這個函數(shù)和上面的函數(shù)很像，不同的是，上面是用索引獲取對應(yīng)值，而這里使用屬性來獲取對應(yīng)值，就像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)

我們有一個學(xué)生的實例的列表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()

一看這個函數(shù)名，就知道是用來統(tǒng)計個數(shù)的~返回的是一個dict,key是各個元素，value是對應(yīng)的個數(shù)。一開始，我并不知道這個函數(shù)，都是這樣統(tǒng)計的：

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

現(xiàn)在，直接一句話：

counter=collections.Counter(list)
print counter

之后獲取各元素的個數(shù)，使用方法同一般的dict。如果增加元素e1,e2和對應(yīng)個數(shù)，直接counter.update({e1=4,e2=3})就可以。

此外，這個Counter不僅用來數(shù)數(shù)，還有好多函數(shù)方便各種操作，例如:

• counter.most_common(n) 返回數(shù)目最多的前n個元素和對應(yīng)個數(shù)
• a.substract(b) 返回一個Counter,Counter a減去Counter b，多的元素個數(shù)為正值，少的元素個數(shù)為負值
• counter.elements() 返回一個element列表，其中每個元素有多少個就重復(fù)多少次
• counter.values() 返回個數(shù)列表，通常配合sum(counter.values())
• counter.clear() 重置counter
• del counter[e]刪除元素e和它的紀錄
• 算數(shù)操作：+和- ，對應(yīng)元素個數(shù)加減；|和&，兩個counter的并集和交集

關(guān)于Python相關(guān)內(nèi)容感興趣的讀者可查看本站專題：《Python函數(shù)使用技巧總結(jié)》、《Python面向?qū)ο蟪绦蛟O(shè)計入門與進階教程》、《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python字符串操作技巧匯總》、《Python編碼操作技巧總結(jié)》及《Python入門與進階經(jīng)典教程》

希望本文所述對大家Python程序設(shè)計有所幫助。

最新評論