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

最近看人家的代碼，發(fā)現了很多python內置的函數和數據結構，可以大大減少我們的代碼量（而且人家是優(yōu)化過的）。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)

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

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()

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

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的并集和交集

關于Python相關內容感興趣的讀者可查看本站專題：《Python函數使用技巧總結》、《Python面向對象程序設計入門與進階教程》、《Python數據結構與算法教程》、《Python字符串操作技巧匯總》、《Python編碼操作技巧總結》及《Python入門與進階經典教程》

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

最新評論