Python 列表生成式 的简单实用

列表生成式

#先列举一个简单的例子
a = [i for i in range(10)]
print(a)
#输出结果: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

还可以使用两层循环，可以生成全排列：

b = [m + n for m in 'ABC' for n in 'XYZ']
print(b)
#输出结果: ['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']

使用if

c = [i for i in range(10) if i > 5]
print(c)
#输出结果: [6, 7, 8, 9]

列表生成式 使用 if …else … 函数 注意顺序和用法

d = [i if i > 5 else i*10 for i in range(10)]
print(d)
#输出结果: [0, 10, 20, 30, 40, 50, 6, 7, 8, 9]

随机取3个不重复的数字

list1= [(i,j,k) for i in [1,2,3] for k in [1,2,3] for j in [1,2,3] if i!=j and j!=k and i!=k]
print(list1)
#输出结果:  [(1, 3, 2), (1, 2, 3), (2, 3, 1), (2, 1, 3), (3, 2, 1), (3, 1, 2)]

Python内置的enumerate函数可以把一个list变成索引-元素对，

返回的结果是索引和元素对

ls = list("abcdabcdabcd")
for i, value in enumerate(ls):
    print(i, value)