python列表推导式中使用if语句及他们的等价形式

python列表推导式中使用if语句及他们的等价形式

在python的列表推导式中使用if语句有两种形式：

# 方式1: 起过滤作用

[expr(x) for x in li if cond(x)]

# 方式1 等价于filter函数

list(map(expr, filter(cond, li)))

 

# 方式2: 作为三元表达式使用，满足条件将其作expr1处理，不满足做expr2处理

[expr1(x) if cond(x) else expr2(x) for x in li]

# 方式2等价于

list(map(lambda x: expr1(x) if cond(x) else expr2(x), li))

# 方式1: 起过滤作用
[expr(x) for x in li if cond(x)]
# 方式1 等价于filter函数
list(map(expr, filter(cond, li)))

# 方式2: 作为三元表达式使用，满足条件将其作expr1处理，不满足做expr2处理
[expr1(x) if cond(x) else expr2(x) for x in li]
# 方式2等价于
list(map(lambda x: expr1(x) if cond(x) else expr2(x), li))

第一种形式和使用filter函数的效果是等价的，这里的if条件相当于过滤条件；

第二种形式跟普通的列表表达式是一样的，使用了一个if-else语句在映射表达式中，if-else语句等价于C语言中的三元表达式。满足if条件执行表达式1(expr1)不满足条件执行表达式2(expr2)。

Examples

# 1-100中除3余1且除5余2的数

p = [x for x in range(1, 101) if x % 3 == 1 and x % 5 == 2]

p = list(filter(lambda x: x%3==1 and x%5==2, range(1, 101)))

 

# 1-1000中奇数去5的余数，偶数取5的倍数

p = [divmod(x, 5)[1] if x % 2 else divmod(x, 5)[0] for x in range(1, 1001)]

p = list(map(lambda x: divmod(x, 5)[1] if x%2 else divmod(x, 5)[0], range(1, 1001)))

# 1-100中除3余1且除5余2的数
p = [x for x in range(1, 101) if x % 3 == 1 and x % 5 == 2]
p = list(filter(lambda x: x%3==1 and x%5==2, range(1, 101)))

# 1-1000中奇数去5的余数，偶数取5的倍数
p = [divmod(x, 5)[1] if x % 2 else divmod(x, 5)[0] for x in range(1, 1001)]
p = list(map(lambda x: divmod(x, 5)[1] if x%2 else divmod(x, 5)[0], range(1, 1001)))