Python tricks(3) -- list和dict的遍歷和方法

每一個人在使用python的過程當中都會遍歷list和dict.

List遍歷

最經常使用最簡單的遍歷list的方法

a = ["a", "b", "c", "d"]

# simple iterate
for i in a:
    print i

可是, 若是我須要拿到list的index, 不少人可能會這樣寫

a = ["a", "b", "c", "d"]

# index & value
for i in xrange(len(a)):
    print i, a[i]

其實, python提供了一個方法enumerate, 用法以下

a = ["a", "b", "c", "d"]

# iterate with index
for i, el in enumerate(a):
    print i, el

上面兩種方式的結果相同

0 a
1 b
2 c
3 d

這是一種更加方便便捷的方式, 雖然少寫不了幾個字符, 從代碼可讀性等方面來考量的話, 仍是清晰不少的.

代碼應該讓人一目瞭然, 目的明確, 若是多種方式能夠實現相同的功能, 那麼咱們應該選擇一種你們更加容易理解的, enumerate就是這樣的方式.

enumerate(iterable[, start]) -> iterator for index, value of iterable

第二個參數在不少時候也是頗有用的, 好比我不但願從0開始, 但願從1開始

a = ["a", "b", "c", "d"]

# iterate with index
for i, el in enumerate(a, 1):
    print i, el

輸出以下

1 a
2 b
3 c
4 d

若是你使用range的話, 會蹩腳不少.

Dict遍歷

dict最簡單的遍歷方式

d = {'a': 1, 'c': 3, 'b': 2, 'd': 4}

for k in d:
    print k

for k in d:
    print k, d[k]

上面遍歷k和v的方式並很差, 顯得很蹩腳. dict自己提供了iteritems()方法, 能夠作到k,v對遍歷.

d = {'a': 1, 'c': 3, 'b': 2, 'd': 4}

# d.viewitems()
for k, v in d.iteritems():
    print k, v

dict還有個viewitems方法, 這個直接看到的是所有k,v對.

iteritems和viewitems的區別是什麼? 能夠類比range和xrange的區別.

你們確定都瞭解range和xrange的區別, 在遍歷的時候儘可能使用xrange, 特別是當遍歷的範圍比較大的時候.

In [1]: print range(10)
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

In [2]: print xrange(10)
xrange(10)

In [3]: print type(xrange(10))
<type 'xrange'>

range是直接返回一個保存全量數據的list, 空間複雜度是O(n), 而xrange是在遍歷中不斷生成的, 遍歷的效率更高, 並且空間複雜度是O(1) (我的理解, 沒看過具體實現).

iteritems和viewitems均可以完成遍歷, 兩者的不一樣用下面的代碼來講明

d = {'a': 1, 'c': 3, 'b': 2, 'd': 4}

# d.viewitems()
for k, v in d.iteritems():
    print k, v

print type(d.viewitems()), type(d.itervalues())
print d.viewitems(), d.itervalues()

輸出以下

a 1
c 3
b 2
d 4
<type 'dict_items'> <type 'dictionary-valueiterator'>
dict_items([('a', 1), ('c', 3), ('b', 2), ('d', 4)]) <dictionary-valueiterator object at 0x103d028e8>

viewitems直接返回的是[('a', 1), ('c', 3), ('b', 2), ('d', 4)], 熟悉dict構造函數的人應該知道, 這也是一種構造dict的方式.

d = dict(zip(("a", "b", "c", "d"), (1, 2, 3, 4)))
# d = {'a': 1, 'c': 3, 'b': 2, 'd': 4}

dict還有其餘幾個方法

In [10]: import itertools

In [11]: ["".join(i) for i in itertools.product(("iter", "view"), ("keys", "values", "items"))]
Out[11]: ['iterkeys', 'itervalues', 'iteritems', 'viewkeys', 'viewvalues', 'viewitems']

dict的完整示例代碼

d = dict(zip(("a", "b", "c", "d"), (1, 2, 3, 4)))
# d = {'a': 1, 'c': 3, 'b': 2, 'd': 4}

for k in d:
    print k

# d.viewkeys()
for k in d.iterkeys():
    print k

print type(d.viewkeys()), type(d.iterkeys())
print d.viewkeys(), d.iterkeys()

# d.viewvalues()
for v in d.itervalues():
    print v

print type(d.viewvalues()), type(d.itervalues())
print d.viewvalues(), d.itervalues()

# d.viewitems()
for k, v in d.iteritems():
    print k, v

print type(d.viewitems()), type(d.itervalues())
print d.viewitems(), d.itervalues()

 

水平有限, 歡迎拍磚!