python列表推导式

# -*- 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']