Python 列表推导式简单使用

无条件列表推导式

li = []
for i in range(5):
    li.append(i)
print(li)

# [0, 1, 2, 3, 4]

li2 = [i for i in range(5)]
print(li2)

# # [0, 1, 2, 3, 4]

有条件子句的列表推导式

li = []
for i in range(5):
    if i % 2 == 0:
        li.append(i)
print(li)

# [0, 2, 4]

li2 = [i for i in range(5) if i % 2 == 0]
print(li2)

[0, 2, 4]

嵌套循环的列表推导式

li = []
for a in range(3):
    for b in range(2):
        li.append((a, b))
print(li)

# [(0, 0), (0, 1), (1, 0), (1, 1), (2, 0), (2, 1)]

li2 = [(a, b) for a in range(3) for b in range(2)]
print(li2)

# [(0, 0), (0, 1), (1, 0), (1, 1), (2, 0), (2, 1)]