在Python中的一行for循环中声明变量[重复]

Question

我有一个像这样的for循环的函数

def my_func():
    order = [1,2,3,4,5,6,7]
    my_list = []
    for o in order:
        tmp = is_odd(o)
        if tmp:
            my_list.append(tmp)
    return my_list

def is_odd(n):
    if n%2 != 0:
        return n
    else:
        return False

我想把它转换成一行 python for 循环。我是这样做的

def my_func():
    order = [1,2,3,4,5,6,7]
    my_list = [is_odd(o) for o in order if is_odd(o)]
    return my_list

def is_odd(n):
    if n%2 != 0:
        return n
    else:
        return False

这段代码的问题是它在每次循环迭代中调用is_odd 函数两次。有没有其他方法可以将循环转换为一行循环并且只调用一次is_odd 函数？

（这不是原始代码。只是一个例子）

Answer 1

如问题答案中所述，我已将其作为重复链接到您的问题：

• Python list comprehension: test function return

您将创建一个生成器（或列表）来生成函数调用的（中间）结果，并基于此生成过滤后的输出：

例子：

def my_func():
    order = [1,2,3,4,5,6,7]
    intermediate_results = (is_odd(o) for o in order)
    # alternatively filter using `filter(None, intermediate_results)`
    return [i for i in intermediate_results if i]

Answer 2

在这种情况下，您可以使用filter function. 例如。 my_list = filter(is_odd, order).

如果你有类似需求，请看itertools module.

Answer 3

嗨——看起来你为o的实例返回了函数is_odd(o)（因此再次运行它），其中函数is_odd(o)返回true。以下应该只运行一次：

def my_func():
    order = [1,2,3,4,5,6,7]
    my_list = [o for o in order if is_odd(o)]
    return my_list