Python 三元表达式、列表推导式、生成器表达式

一、三元表达式

三元表达式，也称为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]