Python list dict iteration

本文主要是介绍Python list dict iteration，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

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

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