本文介绍Python中高效遍历list和dict的方法，包括使用enumerate获取索引值和利用dict的iteritems及viewitems方法进行键值对遍历。
每个人在使用python的过程中都会遍历list和dict.
List遍历
最常用最简单的遍历list的方法
|
1
2
3
4
5
|
a
=
[
"a"
,
"b"
,
"c"
,
"d"
]
# simple iterate
for
i
in
a:
print
i
|
但是, 如果我需要拿到list的index, 很多人可能会这样写
|
1
2
3
4
5
|
a
=
[
"a"
,
"b"
,
"c"
,
"d"
]
# index & value
for
i
in
xrange
(
len
(a)):
print
i, a[i]
|
其实, python提供了一个方法enumerate, 用法如下
|
1
2
3
4
5
|
a
=
[
"a"
,
"b"
,
"c"
,
"d"
]
# iterate with index
for
i, el
in
enumerate
(a):
print
i, el
|
上面两种方式的结果相同
|
1
2
3
4
|
0
a
1
b
2
c
3
d
|
这是一种更加方便便捷的方式, 虽然少写不了几个字符, 从代码可读性等方面来考量的话, 还是清晰很多的.
代码应该让人一目了然, 目的明确, 如果多种方式可以实现相同的功能, 那么我们应该选择一种大家更加容易理解的, enumerate就是这样的方式.
|
1
|
enumerate
(iterable[, start])
-
> iterator
for
index, value of iterable
|
第二个参数在很多时候也是很有用的, 比如我不希望从0开始, 希望从1开始
|
1
2
3
4
5
|
a
=
[
"a"
,
"b"
,
"c"
,
"d"
]
# iterate with index
for
i, el
in
enumerate
(a,
1
):
print
i, el
|
输出如下
|
1
2
3
4
|
1
a
2
b
3
c
4
d
|
如果你使用range的话, 会蹩脚很多.
Dict遍历
dict最简单的遍历方式
|
1
2
3
4
5
6
7
|
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对遍历.
|
1
2
3
4
5
|
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, 特别是当遍历的范围比较大的时候.
|
1
2
3
4
5
6
7
8
|
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都可以完成遍历, 二者的不同用下面的代码来说明
|
1
2
3
4
5
6
7
8
|
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()
|
输出如下
|
1
2
3
4
5
6
|
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的方式.
|
1
2
|
d
=
dict
(
zip
((
"a"
,
"b"
,
"c"
,
"d"
), (
1
,
2
,
3
,
4
)))
# d = {'a': 1, 'c': 3, 'b': 2, 'd': 4}
|
dict还有其他几个方法
|
1
2
3
4
|
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的完整示例代码
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
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()
|
扫一扫
844
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
