Python列表推导式

列表推导式

• 列表推导式使用非常简洁的方式来快速生成满足特定需求的列表，代码具有非常强的可读性。

• 语法形式为：

  [expression for expr1 in sequence1 if condition1

  for expr2 in sequence2 if condition2

  ​ for expr3 in sequence3 if condition3

  ​ ...

  ​ for exprN in sequenceN if conditionN]

• 列表推导式在逻辑上等价于一个循环语句，只是形式上更加简洁。

  例如：

  aList = [x*x for x in range(10)]

  相当于

  >>> aList = []

  >>> for x in range(10):

  ​ aList.append(x*x)

• 实现嵌套列表的平铺

  >>> vec = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

  [num for elem in vec for num in elem]

  [1, 2, 3, 4, 5, 6, 7, 8, 9]

  在这个列表推导式中有2个循环，其中第一个循环可以看作是外循环，执行的慢；而第二个循环可以看作是内循环，执行的快。上面代码的执行过程等价于下面的写法：

  >>> vec = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

  >>> result = []

  >>> for elem in vec:

  ​ for num in elem:

  ​ result.append(num)

  >>> result

  [1, 2, 3, 4, 5, 6, 7, 8, 9]

• 过滤不符合条件的元素

  在列表推导式中可以使用if子句对列表中的元素进行筛选，只在结果列表中保留符合条件的元素。下面的代码可以列出当前文件夹下所有Python源文件：

  >>> import os

  >>> [filename for filename in os.listdir('.') if filename.endswith(('.py', '.pyw'))]

  下面的代码用于从列表中选择符合条件的元素组成新的列表：

  >>> aList = [-1, -4, 6, 7.5, -2.3, 9, -11]

  >>> [i for i in aList if i>0] #所有大于0的数字

  [6, 7.5, 9]

• 同时遍历多个列表或可迭代对象

  [(x, y) for x in [1, 2, 3] for y in [3, 1, 4] if x != y]

  [(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

  >>> [(x, y) for x in [1, 2, 3] if x==1 for y in [3, 1, 4] if y!=x]

  [(1, 3), (1, 4)]

  对于包含多个循环的列表推导式，一定要清楚多个循环的执行顺序或“嵌套关系”。例如，上面第一个列表推导式等价于

  >>> result = []

  >>> for x in [1, 2, 3]:

  ​ for y in [3, 1, 4]:

  ​ if x != y:

  ​ result.append((x,y))

  >>> result

  [(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

• 列表推导式中可以使用函数或复杂表达式

  def f(v):

  ​ if v%2 == 0:

  ​ v = v**2

  ​ else:

  ​ v = v+1

  ​ return v

  >>> [f(v) for v in [2, 3, 4, -1] if v>0]

  [4, 4, 16]

  >>> [v**2 if v%2 == 0 else v+1 for v in [2, 3, 4, -1] if v>0]

  [4, 4, 16]

  >>> x = list(range(10))

  >>> [item>5 for item in x]

  [False, False, False, False, False, False, True, True, True, True]