python列表推导式

摘要: 看到python列表推导式，感觉非常强大，写了一些小例子。列表推导式为从序列中创建列表提供了一个简单的方法。列表推导式由包含一个表达式的括号组成，表达式后面跟随一个 for 子句，之后可以有零或多个 for 或 if 子句。结果是一个列表，由表达式依据其后面的 for 和 if 子句上下文计算而来的结果构成。

# -*- coding:utf-8 -*-
#列表推导式
L = [-2, -1, 0, 1, 2]
a = [[i, i**2] for i in L if i>0 ]
b = [abs(i) for i in L]
c = [(x, x**2) for x in range(6)]

vec = [[1,2,3], [4,5,6], [7,8,9]]
d = [i for elem in vec for i in elem]

person = ['  shim', '  1991-11-11 ', 'shanghai  ']
e = [weapon.strip() for weapon in person]

print a
print b
print c
print d
print e

结果

[[1, 1], [2, 4]]
[2, 1, 0, 1, 2]
[(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25)]
[1, 2, 3, 4, 5, 6, 7, 8, 9]
['shim', '1991-11-11', 'shanghai']