1. 程式人生 > >Python 列表生成式 的簡單實用

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)