python基礎 函式之 強大的zip

pythontab發表於2013-03-15

一、程式碼引導

首先看這一段程式碼:

>>> name=('jack','beginman','sony','pcky')
>>> age=(2001,2003,2005,2000)
>>> for a,n in zip(name,age):
    print a,n 

輸出:

jack 2001

beginman 2003

sony 2005

pcky 2000

再看這一段程式碼:

all={"jack":2001,"beginman":2003,"sony":2005,"pcky":2000}
 for i in all.keys():
     print i,all[i]

 輸出:

sony 2005

pcky 2000

jack 2001

beginman 2003

發現它們之間的區別麼?

最顯而易見的是:第一種簡潔、靈活、而且能順序輸入。

二、zip()函式

它是Python的內建函式,(與序列有關的內建函式有:sorted()、reversed()、enumerate()、zip()),其中sorted()和zip()返回一個序列(列表)物件,reversed()、enumerate()返回一個迭代器(類似序列)

>>> type(sorted(s))
 <type 'list'>
 >>> type(zip(s))
 <type 'list'>
 >>> type(reversed(s))
 <type 'listreverseiterator'>
 >>> type(enumerate(s))
 <type 'enumerate'>

那麼什麼是zip()函式 呢?


我們help(zip)看看:

>>> help(zip)

Help on built-in function zip in module __builtin__:

zip(...)

    zip(seq1 [, seq2 [...]]) -> [(seq1[0], seq2[0] ...), (...)]

   

    Return a list of tuples, where each tuple contains the i-th element

    from each of the argument sequences.  The returned list is truncated

    in length to the length of the shortest argument sequence.

提示:不懂的一定多help


定義:zip([seql, ...])接受一系列可迭代物件作為引數,將物件中對應的元素打包成一個個tuple(元組),然後返回由這些tuples組成的list(列表)。若傳入引數的長度不等,則返回list的長度和引數中長度最短的物件相同。

>>> z1=[1,2,3]
 >>> z2=[4,5,6]
 >>> result=zip(z1,z2)
 >>> result
 [(1, 4), (2, 5), (3, 6)]
 >>> z3=[4,5,6,7]
 >>> result=zip(z1,z3)
 >>> result
 [(1, 4), (2, 5), (3, 6)]
 >>>

zip()配合*號運算子,可以將已經zip過的列表物件解壓

>>> zip(*result)

[(1, 2, 3), (4, 5, 6)]

更近一層的瞭解:

* 二維矩陣變換(矩陣的行列互換)

比如我們有一個由列表描述的二維矩陣

a = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

透過python列表推導的方法,我們也能輕易完成這個任務

print [ [row[col] for row in a] for col in range(len(a[0]))]

[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

另外一種讓人困惑的方法就是利用zip函式:

>>> a = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

>>> zip(*a)

[(1, 4, 7), (2, 5, 8), (3, 6, 9)]

>>> map(list,zip(*a))

[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

zip函式接受任意多個序列作為引數,將所有序列按相同的索引組合成一個元素是各個序列合併成的tuple的新序列,新的序列的長度以引數中最短的序列為準。另外(*)運算子與zip函式配合可以實現與zip相反的功能,即將合併的序列拆成多個tuple。

①tuple的新序列

>>>>x=[1,2,3],y=['a','b','c']

>>>zip(x,y)

[(1,'a'),(2,'b'),(3,'c')]


②新的序列的長度以引數中最短的序列為準.

>>>>x=[1,2],y=['a','b','c']

>>>zip(x,y)

[(1,'a'),(2,'b')]


③(*)運算子與zip函式配合可以實現與zip相反的功能,即將合併的序列拆成多個tuple。

>>>>x=[1,2,3],y=['a','b','c']

>>>>zip(*zip(x,y))

[(1,2,3),('a','b','c')]


相關文章