python的嵌套列表推导

连接： http://blog.lerner.co.il/understanding-nested-list-comprehensions-in-python/

列表推导的核心点就是： 迭代器

Demo 1 获取linux 保存密码文件的每一行的字符数

（1） 常规实现

>>> ret = []
>>> for line in open('/etc/passwd'):
...     ret.append(len(line))

（2）列表推导实现

>>> [len(line) for line in open('/etc/passwd')]

其实line是一个字符串，string对象，在python里面也是可以迭代的。所以可以嵌套跌套，不过在嵌套推导前，先看一个小例子。

Demo 2

>>> [(x,y) for x in range(5) for y in range(5)]
[(0, 0), (0, 1), (0, 2), (0, 3), (0, 4), (1, 0), (1, 1), (1, 2), (1, 3), (1, 4), (2, 0), (2, 1), (2, 2), (2, 3), (2, 4), (3, 0), (3, 1), (3, 2), (3, 3), (3, 4), (4, 0), (4, 1), (4, 2), (4, 3), (4, 4)]

这个的常规实现就是：

>>> ret = []
>>> for x in range(5):
...     for y in range(5):
...             ret.append((x, y))

当然如果把上面的列表推导重新展示一下，如下：

>>> [(x, y)
... for x in range(5)
... for y in range(5)]

这下应该清晰很多了。我们来分解一下，具体是什么意思。

• 我们的输出是一个tuple （x, y）. 也就是说这个列表表达式将会产生一个二维tuple的list
  
• 先手遍历range（5），对x分别赋值0-4
  
• 然后对于每个x值，又遍历range（5），对y分别赋值0-4
  
• 值的数量取决取第二个for的遍历值的数量
  
• 输出是（0,0）-（4,4）的二维tuple

我们可以稍微修改一下

>>> [(x, y)
... for x in range(5)
... for y in range(x + 1)]

这里，第二个for，对y赋值取的是range（x + 1），也就是y小于等于x

Demo 3 获取选手的平均值

基本信息如下：

>>> scores = {'Reuven':[300, 250, 350, 400], 
...  'Atara':[200, 300, 450, 150], 
...  'Shikma':[250, 380, 420, 120], 
...  'Amotz':[100, 120, 150, 180] }

要求： 获取每个人的平均值

（1）常规方法

ret = dict()
>>> for name, score in scores.items():
...     ret[name] = sum(score) / len(score)

（2）列表推导

>>> def average(input):
...     return sum(input) / len(input)    

>>> dict([(name, average(score)) for name, score in scores.items()])
{'Reuven': 325, 'Atara': 275, 'Amotz': 137, 'Shikma': 292}

如果你使用的python2.7或者3.1，上面的话，就不用转换，因为已经支持了字典推导式（Dictionary comprehensions）

>>> {name: average(score) for name, score in scores.items()}

如果想要获取所有人的平均值怎么办：

（1）常规方法:

>>> sum = 0
>>> plays_num = 0
>>> for one_player_score in scores.values():
...     for score in one_player_score:
...             sum += score
...             plays_num += 1
... 
>>> avg = sum / plays_num

（2）列表推导

>>> average([ one_score  
...               for one_player_scores in scores.values()  
...               for one_score in one_player_scores ])
257

是不是很简洁。

如果想要查大于200分的所有分数

>>> [ one_score      
...       for one_player_scores in scores.values()     
...       for one_score in one_player_scores
...       if one_score > 200]
[300, 250, 350, 400, 300, 450, 250, 380, 420]

Demo 4

现在有一个酒店的住房信息，是一个list，每个list元素也是list，代表房间住房信息，list里面是字典，存放住过房客的信息。如下：

>>> rooms = [[{'age': 14, 'hobby': 'horses', 'name': 'A'},  
           {'age': 12, 'hobby': 'piano', 'name': 'B'},  
           {'age': 9, 'hobby': 'chess', 'name': 'C'}],  
          [{'age': 15, 'hobby': 'programming', 'name': 'D'}, 
           {'age': 17, 'hobby': 'driving', 'name': 'E'}],  
          [{'age': 45, 'hobby': 'writing', 'name': 'F'},  
           {'age': 43, 'hobby': 'chess', 'name': 'G'}]]
           

• 获取酒店房客的名字

  >>> [ person['name']
  ...     for room in rooms
  ...     for person in room]
  ['A', 'B', 'C', 'D', 'E', 'F', 'G']

• 获取喜欢下棋的人

  >>> [ person['name']
  ...     for room in rooms
  ...     for person in room
  ...     if person['hobby'] == 'chess' ]

本文只是一个学习笔记，如果了解详细，还请看文章开头的连接。