Python list dict iteration-白红宇

Python list dict iteration

阅读量：4030 次

发布时间：2019-05-24

本文共 3101 字，大约阅读时间需要 10 分钟。

P

2014-01-23 23:11 by 会被淹死的鱼, 1631 阅读, 0 评论, ,

每个人在使用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
         
       

上面两种方式的结果相同

这是一种更加方便便捷的方式, 虽然少写不了几个字符, 从代码可读性等方面来考量的话, 还是清晰很多的.

代码应该让人一目了然, 目的明确, 如果多种方式可以实现相同的功能, 那么我们应该选择一种大家更加容易理解的, 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
         
       

输出如下

如果你使用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()
         
       

转载地址：http://glhbi.baihongyu.com/

你可能感兴趣的文章