1. 程式人生 > >Python 三元表達式、列表推導式、生成器表達式

Python 三元表達式、列表推導式、生成器表達式

urn obj 推導 變量 中間 用法 tor lis 元表

一、三元表達式

三元表達式,也稱為if,else的緊湊形式。具體用法如下:

def max(a,b):
    if a>b:
        return a
    else:
        return b


def max(a,b):
    return a if a>b else b  

上面這兩個函數實現的功能是完全一樣的。第二個函數中 a if a>b else b 就是一個三元表達式。 左邊是條件為真時返回的值,中間是判斷條件,右邊是條件假返回的值。

二、列表推導式

大致的格式如下 [表達式 for 變量 in 列表 if 條件] ,其中if條件是可選的。

我們先來看看列表推導式的一些簡單例子:

print([i**2 for i in range(10)])
#[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

print([i**2 for i in range(10) if i>=5] )
#[25, 36, 49, 64, 81]

print([x*y for x in [1,2,3] for y in [4,5,6]])
#[4, 5, 6, 8, 10, 12, 12, 15, 18]

其實既然有列表推導式,那麽同樣也有集合推導式,字典推導式

print({x*y for x in [1,2,3] for
y in [4,5,6]}) #{4, 5, 6, 8, 10, 12, 15, 18} #實現字典的key和value快速交換: dic1={a:1,b:2} dic2={v: k for k, v in dic1.items()} print(dic2) #{1: ‘a‘, 2: ‘b‘}

三、生成器表達式

將列表推導式的[]換成()就變成生成器表達式了。

a=(i*i for i in range(10))
print(a)          #<generator object <genexpr> at 0x027DE180>
print(list(a))  #
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

Python 三元表達式、列表推導式、生成器表達式