Python列表生成式及扩展

• 初始化一个列表

my_lis = list(range(1,11))
print(my_lis)
# [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

• 使用列表生成式生成新的列表

my_lis1 = [x*(x+1) for x in my_lis]
print([x*(x+1) for x in my_lis])
# [2, 6, 12, 20, 30, 42, 56, 72, 90, 110]
# 取出my_lis1中大于50的索引值
print([i for i,x in enumerate(my_lis1) if x >50])
# [6, 7, 8, 9]

• 使用if语句的列表生成式

my_lis2 = [x for x in my_lis1 if x>25]
print(my_lis2)
# [30, 42, 56, 72, 90, 110]
my_lis3 = [x if x>25 else 0 for x in my_lis1 ]
print(my_lis3)
# [0, 0, 0, 0, 30, 42, 56, 72, 90, 110]

• 两个列表结合的列表生成式

my_lis4 = [x+y for x,y in zip(my_lis,my_lis1)]
print(my_lis4)
# [3, 8, 15, 24, 35, 48, 63, 80, 99, 120]

• 双重循环的列表生成式

my_array=[[1,2,3],[4,5,6],[7,8,9]]
print([y for x in my_array  for y in x])
# [1, 2, 3, 4, 5, 6, 7, 8, 9]
print([y for x in my_array  for y in x if y>4])
# [5, 6, 7, 8, 9]

• 生成多维列表

my_array1 = [list(range(x)) for x in range(1,5)]
print(my_array1)
# [[0], [0, 1], [0, 1, 2], [0, 1, 2, 3]]