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

水平有限, 欢迎拍砖!