我有一个包含字符串和None值的列表xs。如何使用列表理解来调用每个字符串上的函数,但将None值转换为“”(而不是将它们传递给函数)?

我尝试了:

[f(x) for x in xs if x is not None else '']

但它给出了SyntaxError。正确的语法是什么?


如果您试图创建一个省略基于条件的值的列表理解,请参阅带条件的列表理解。

如果需要考虑两个以上的条件结果,请注意Python的条件表达式不支持elif。相反,需要嵌套if/else条件。有关详细信息,请参见列表理解条件中的“elif”。


当前回答

单向:

def change(x):
    if x is None:
        return f(x)
    else:
        return ''

result = [change(x) for x in xs]

尽管你有:

result = map(change, xs)

或者可以使用lambda内联。

其他回答

其他解决方案对于单个if/else构造非常有用。然而,列表理解中的三元语句很难理解。

使用函数有助于可读性,但在映射是输入的工作流中,这样的解决方案很难扩展或适应。词典可以缓解这些担忧:

xs = [None, 'This', 'is', 'a', 'filler', 'test', 'string', None]

d = {None: '', 'filler': 'manipulated'}

res = [d.get(x, x) for x in xs]

print(res)

['', 'This', 'is', 'a', 'manipulated', 'test', 'string', '']

这与如何理解列表有关。

请记住以下几点:

[ expression for item in list if conditional ]

相当于:

for item in list:
    if conditional:
        expression

如果表达式的格式稍有不同(请考虑在句子中切换主语和动词的顺序)。

因此,您的代码[x+1 for x in l if x>=45]执行以下操作:

for x in l:
    if x >= 45:
        x+1

然而,此代码[x+1 if x>=45 else x+5 for x in l]执行此操作(重新排列表达式后):

for x in l:
    if x>=45: x+1
    else: x+5

让我们用这个问题来回顾一些概念。我认为最好先看看基本面,这样你就可以推断出不同的情况。

其他答案为您的问题提供了具体答案。我将首先介绍一些一般情况,然后回答这个问题。

基本原理

列表理解中的if/else语句涉及两件事:

列表解析条件表达式(三元运算符)

1.列出理解

它们提供了创建列表的简洁方法。

它的结构包括:“括号包含一个表达式,后跟一个for子句,然后是零个或多个for或if子句”。

案例1

这里我们没有条件。iterable中的每个项都添加到new_list中。

new_list = [expression for item in iterable]
new_list = [x for x in range(1, 10)]
> [1, 2, 3, 4, 5, 6, 7, 8, 9]

案例2

这里我们有一个条件。

示例1

条件:只有偶数将被添加到new_list中。

new_list = [expression for item in iterable if condition == True]
new_list = [x for x in range(1, 10) if x % 2 == 0]
> [2, 4, 6, 8]

示例2

条件:只有3的倍数的偶数才会添加到new_list中。

new_list = [expression for item in iterable if condition == True]
new_list = [x for x in range(1, 10) if x % 2 == 0 if x % 3 == 0]
> [6]

但是,如果在new_list中使用两个if,为什么会有一个条件?

前面的表达式可以写成:

new_list = [x for x in range(1, 10) if x % 2 and x % 3 == 0]
> [6]

我们只使用一个if语句。

这就像在做:

new_list = []
for x in range(1, 10):
    if x % 2 == 0 and x % 3 == 0:
        new_list.append(x)
> [6]

示例3

为了便于讨论,您也可以使用或。

条件:偶数或3的倍数将添加到new_list中。

new_list = [x for x in range(1, 10) if x % 2 == 0 or x % 3 == 0]
> [2, 3, 4, 6, 8, 9]

案例3

多个条件:

这里我们需要条件表达式(三元运算符)的帮助。

2.条件表达式

什么是条件表达式?名称说明:具有某些条件的Python表达式。

<Exp1> if condition else <Exp2>

首先评估条件。如果条件为True,则计算并返回<Exp1>。如果条件为False,则求值并返回<Exp2>。

具有多个条件的条件表达式:

<Exp1> if condition else <Exp2> if condition else <Exp3>...    

Real Python的一个示例:

age = 12
s = 'minor' if age < 21 else 'adult'
> minor

s的值取决于年龄值。

3.列出带条件的理解

我们像这样把列表理解和条件放在一起。

new_list = [<Conditional Expression> for <item> in <iterable>]

new_list = [<Exp1> if condition else <Exp2> if condition else <Exp3> for <item> in <iterable>]

条件:偶数将被添加为“偶数”,第三个数字将添加为“第三个”,其余数字将添加“奇数”。

new_list = ['even' if x % 2 == 0 else 'number three' if x == 3 else 'odd' 
             for x in range(1, 10)]
> ['odd', 'even', 'number three', 'even', 'odd', 'even', 'odd', 'even', 'odd']

问题的答案

[f(x) for x in xs if x is not None else '']

这里我们对列表的结构有一个问题:xs中的for x应该位于表达式的末尾。

正确方式:

[f(x) if x is not None else '' for x in xs]

进一步阅读:

Python是否有三元条件运算符?

具体的问题已经在前面的答案中解决了,所以我将阐述在列表理解中使用条件的一般思想。

下面是一个例子,说明如何在列表理解中编写条件句:

X = [1.5, 2.3, 4.4, 5.4, 'n', 1.5, 5.1, 'a']     # Original list

# Extract non-strings from X to new list
X_non_str = [el for el in X if not isinstance(el, str)]  # When using only 'if', put 'for' in the beginning

# Change all strings in X to 'b', preserve everything else as is
X_str_changed = ['b' if isinstance(el, str) else el for el in X]  # When using 'if' and 'else', put 'for' in the end

注意,在X_non_str的第一个列表理解中,顺序是:

可迭代if条件中项的表达式

在X_str_changed的最后一个列表理解中,顺序为:

表达式1 if条件else表达式2 for可迭代项

我总是觉得很难记住表达式1必须在if之前,表达式2必须在else之后。我的头脑希望两者都在之前或之后。

我想它是这样设计的,因为它类似于正常语言,例如“如果下雨,我想呆在里面,否则我想出去”

在普通英语中,上述两种类型的列表理解可以表述为:

仅当:

如果苹果是管道,则在苹果盒中提取苹果

和if/else

如果apple-is_pipe,则标记apple,否则在apple_box中为apple-it_unmark

从可迭代的

最好先概括所有可能的形式,而不是给出问题的具体答案。否则,读者将不知道答案是如何确定的。下面是我在决定最后一个形式中是否可以使用final else子句之前想出的几个通用形式。

[expression1(item)                                        for item in iterable]

[expression1(item) if conditional1                        for item in iterable]

[expression1(item) if conditional1 else expression2(item) for item in iterable]

[expression1(item) if conditional1 else expression2(item) for item in iterable if conditional2]

item的值不需要在任何条件子句中使用。conditional3可以用作向输出列表添加或不添加值的开关。

例如,要创建一个新列表,从原始字符串列表中删除空字符串或空白字符串:

newlist = [s for s in firstlist if s.strip()]