Python 列表生成式 的簡單實用
阿新 • • 發佈:2018-12-17
列表生成式
#先列舉一個簡單的例子
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)