格式:
格式一 : [i for i in 可迭代的]
list1 = [i for i in range(1, 21)]
print(list1)
# 输出结果[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20]
list1 = [i + 2 for i in range(1, 10)]
print(list1)
# 输出结果[3, 4, 5, 6, 7, 8, 9, 10, 11]
格式二 : [i for i in 可迭代的 if 条件]
list1 = [i for i in range(0, 10) if i % 2 == 0]
print(list1)
# 输出结果为[0, 2, 4, 6, 8]
list1 = ['1', 'abc', '2', 'hello', '4', 'world']
list2 = [world for world in list1 if world.isalpha()]
print(list2)
# 输出结果为['abc', 'hello', 'world']
格式三 : [结果1 if 条件 else 结果2 for 变量 in 可迭代的]
# 如果h开头的 首字母大写,不是h开头的全部转换成大写
list1 = ['1', 'abc', '2', 'hello', '4', 'world']
list2 = [world if world.startswith('h') else world.upper() for world in list1]
print(list2)
# 输出结果为['1', 'ABC', '2', 'hello', '4', 'WORLD']
其它
其实还有两个for循环和三个for循环的列表推导式。
# 两个for循环
list1 = [(i, j) for i in range(1, 3) for j in range(1, 3)]
print(list1)
# 输出结果[(1, 1), (1, 2), (2, 1), (2, 2)]
# 三个for循环
list1 = [(i, j, n) for i in range(1, 3) for j in range(1, 3) for n in range(1, 3)]
print(list1)
# 输出结果[(1, 1, 1), (1, 1, 2), (1, 2, 1), (1, 2, 2), (2, 1, 1), (2, 1, 2), (2, 2, 1), (2, 2, 2)]