列表生成式即 List Comprehensions。
-
如果要生成 list[1, 2, 3, 4, 5],那么可以使用如下代码:
# 范围包头不包尾 list(range(1, 6))
但如果要生成 [1×1, 2×2, 3×3, 4×4, 5×5],有两种办法。办法一就是通过循环:
# 通过循环的方式生成 [1×1, 2×2, 3×3, 4×4, 5×5] L = [] for x in range(1, 6): L.append(x * x)
办法二就是通过列表生成式,代码如下:
# 通过列表生成式的方式生成 [1×1, 2×2, 3×3, 4×4, 5×5] [x * x for x in range(1, 6)]
-
列表生成式更厉害的地方是可以在后面添加 if 语句加以判断。例如要得到个连续偶数平方的 list,即 [2×2, 4×4, 6×6, 8×8, 10×10],可以通过如下代码获得:
# 连续偶数平方的 list,即 [2×2, 4×4, 6×6, 8×8, 10×10] [x * x for x in range(1, 11) if x % 2 == 0]
-
同时还能使用双重循环。例如,需要求两个list种任意两个数之和,那么即可使用双重循环,代码如下:
# 两个list种任意两个数之和 L1 = [1, 3, 5] L2 = [2, 4, 6] [x + y for x in L1 for y in L2] # 结果如下 # [3, 5, 7, 5, 7, 9, 7, 9, 11]
当然也有三层和三层以上的循环,但是都很少使用。