每個人在使用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()
水平有限, 歡迎拍磚!