Python – 下一个函数 – 列表理解

Question

我试图理解为什么以紧凑风格编写的代码有效，而以更冗长的风格编写的“相同的代码”不起作用。

此代码产生正确答案：

which_days = [5, 12, 19, 26]
buy_day = next((i for i in which_days if i in range(13, 20)), None)
print(buy_day)
>>> 19 <- correct answer

这段代码确实产生了一对括号：

which_days = [5, 12, 19, 26]
buy_day = []
for i in which_days:
    if i in range((13, 20), None):
        next(which_days)
print(buy_day)
>>>[] <- This is a pair of brackets

Answer 1

这是正确代码的长格式：

which_days = [5, 12, 19, 26]
def _tmp_variable():
    for i in which_days:
        if i in range(13, 20):
            yield i
buy_day = next(_tmp_variable(), None)
del _tmp_variable

不正确的代码不起作用，因为它正在做一些完全不同的事情（更不用说非法；您不能在 Python 3 的列表中调用 next()）。

这是一种不同的长形式，它急切地构建整个列表，而不是一次懒惰地产生一个元素：

which_days = [5, 12, 19, 26]
_tmp_variable = []
for i in which_days:
    if i in range(13, 20):
        _tmp_variable.append(i)
buy_day = _tmp_variable[0] if _tmp_variable else None
del _tmp_variable

如您所见，这是一种浪费，因为它构建了一个完整的列表，然后只使用第一个元素。幸运的是，您的代码的简短版本不会这样做。