python學習：陣列之列表推導

十五樓亮哥發表於2015-01-29
Python陣列

先來看一個例子：取陣列中的偶數的一般做法

array  = [1,2,3,4,5,6,7]
result = []
for k in array:
    if k % 2 == 0:
        result.append(k)
print result

如果使用列表推導的話，實現起來很簡單：
array  = [1,2,3,4,5,6,7]

result = [i for i in array if i % 2 == 0]
print result

再舉個例子，如果想把陣列中每個元素都乘以2的話：
array  = [1,2,3,4,5,6,7]
result = [i * 2 for i in array]
print result

列表推導式書寫形式：　　

[表示式 for 變數 in 列表]    或者  [表示式 for 變數 in 列表 if 條件]


相關文章