python——列表推導式

yucheng1998發表於2019-08-02

列表推導式書寫形式:

[表示式 for 變數 in 列表] 或者 [表示式 for 變數 in 列表 if 條件] 列表推導式(又稱列表解析式)提供了一種簡明扼要的方法來建立列表。

它的結構是在一箇中括號裡包含一個表示式,然後是一個for語句,然後是 0 個或多個 for 或者 if 語句。那個表示式可以是任意的,意思是你可以在列表中放入任意型別的物件。返回結果將是一個新的列表,在這個以 if 和 for 語句為上下文的表示式執行完成之後產生。

列表推導式的執行順序:各語句之間是巢狀關係,左邊第二個語句是最外層,依次往右進一層,左邊第一條語句是最後一層。

[x*y for x in range(1,5) if x > 2 for y in range(1,4) if y < 3]
複製程式碼

他的執行順序是:

for x in range(1,5)
    if x > 2
        for y in range(1,4)
            if y < 3
                x*y
複製程式碼

舉例說明:

    #!/usr/bin/python
    # -*- coding: utf-8 -*-
    
    li = [1,2,3,4,5,6,7,8,9]
    print ([x**2 for x in li])
    [1, 4, 9, 16, 25, 36, 49, 64, 81]
    print ([x**2 for x in li if x>5])
    [36, 49, 64, 81]
    print (dict([(x,x*10) for x in li]))
    {1: 10, 2: 20, 3: 30, 4: 40, 5: 50, 6: 60, 7: 70, 8: 80, 9: 90}
    
    print ([ (x, y) for x in range(10) if x % 2 if x > 3 for y in range(10) if y > 7 if y != 8 ])
    [(5, 9), (7, 9), (9, 9)]
    vec=[2,4,6]
    vec2=[4,3,-9]
    sq = [vec[i]+vec2[i] for i in range(len(vec))]
    print (sq)
    [6, 7, -3]
    print ([x*y for x in [1,2,3] for y in  [1,2,3]])
    [1, 2, 3, 2, 4, 6, 3, 6, 9]
    testList = [1,2,3,4]
    def mul2(x):
        return x*2
    print ([mul2(i) for i in testList])
    [2, 4, 6, 8]
複製程式碼

相關文章