Python中有趣的列表推导式

list1 = [[1, 2], [3, 4], [5, 6], [7, 8]]
list2 = [i for a in list1 for i in a]
print(list2)
# [1, 2, 3, 4, 5, 6, 7, 8]
# 将列表转成numpy矩阵，通过numpy的flatten()方法实现上面的结果
import numpy as np
list1 = [[1, 2], [3, 4], [5, 6]]
list2 = np.array(list1).flatten().tolist()
print(list2)

# 列表推导式配合切片使用
list3 = [[x for x in range(1, 100)][i: i + 3] for i in range(0, 100, 3)]
print(list3)
# [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12], [13, 14, 15], [16, 17, 18], [19, 20, 21], [22, 23, 24], [25, 26, 27], [28, 29, 30], [31, 32, 33], [34, 35, 36], [37, 38, 39], [40, 41, 42], [43, 44, 45], [46, 47, 48], [49, 50, 51], [52, 53, 54], [55, 56, 57], [58, 59, 60], [61, 62, 63], [64, 65, 66], [67, 68, 69], [70, 71, 72], [73, 74, 75], [76, 77, 78], [79, 80, 81], [82, 83, 84], [85, 86, 87], [88, 89, 90], [91, 92, 93], [94, 95, 96], [97, 98, 99], []]
# 不用切片的方式实现上面的结果
list4 = [[i, i + 1, i + 2] for i in range(1, 100, 3)]
print(list4)
# [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12], [13, 14, 15], [16, 17, 18], [19, 20, 21], [22, 23, 24], [25, 26, 27], [28, 29, 30], [31, 32, 33], [34, 35, 36], [37, 38, 39], [40, 41, 42], [43, 44, 45], [46, 47, 48], [49, 50, 51], [52, 53, 54], [55, 56, 57], [58, 59, 60], [61, 62, 63], [64, 65, 66], [67, 68, 69], [70, 71, 72], [73, 74, 75], [76, 77, 78], [79, 80, 81], [82, 83, 84], [85, 86, 87], [88, 89, 90], [91, 92, 93], [94, 95, 96], [97, 98, 99]]

# 嵌套列表推导式,多个并列条件
list_w = [[x for x in range(g - 3, g)] for g in range(22) if g % 3 == 0 and g != 0]
print(list_w)
# [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11], [12, 13, 14], [15, 16, 17], [18, 19, 20]]

总结: 列表推导式供能很强大, 很简洁, 还可以配合lambda匿名函数使用, 有兴趣可以深入研究