列表理解和 len() 与简单的 for 循环

Question

我应该获取一个单词列表，并计算其中长度为 2 个或更多字符且第一个字符和最后一个字符相等的所有单词。

我想出了两种可能的解决方案：

result = 0
for word in words:
    if len(word) >= 2 and word[0] == word[-1]:
        result += 1
return result

vs.

return len([word for word in words if len(word) >= 2 and word[0] == word[-1]])

哪一种是首选解决方案？或者还有更好的吗？

Answer 1

在你的第二个例子中，如果你的 生成器表达式 会比 list-comp 更好名单很大。

sum(1 for word in words if len(word) >= 2 and word[0] == word[-1])

Answer 2

第一个肯定是 Python 中的首选解决方案。

不要忘记你的 Python 之禅：

Python 之禅，作者：Tim Peters

美丽总比丑陋好。

显式优于隐式。

简单总比复杂好。

复杂总比复杂好。

扁平比嵌套更好。

稀疏优于密集。

可读性很重要。

特殊情况还不够特殊
违反规则。

尽管实用性胜过纯粹性。

错误永远不应该悄无声息地过去。

除非明确沉默。

面对歧义，拒绝
猜测的诱惑。

应该有一个——而且最好是
只有一种显而易见的方法可以做到这一点。

虽然这种方式可能并不明显
首先除非你是荷兰人。

现在总比没有好。

尽管从来没有比
现在。

如果实施起来有困难
解释一下，这是一个坏主意。

如果实现很容易
解释一下，这可能是个好主意。

命名空间是一个非常棒的主意
-- 让我们做更多这样的事情！

除此之外，你的解决方案都很好。

Answer 3

我个人认为显式循环更具可读性，但这很大程度上取决于品味（有些人通常更喜欢较短的代码，尤其是当他们必须编写代码时）。

任一版本都可以进一步缩短/改进：

result = 0
for word in words:
    result += int(len(word) >= 2 and word[0] == word[-1])
return result

严格来说 int() 转换是不必要的，因为 True 是一种 1，但它可能更有可读性。同样的方法可以应用于理解：

return sum(len(word) >= 2 and word[0] == word[-1] for word in words)

如果您想使用 len()，我会向读者指出这些值并不重要：

len(1 for word in words if len(word) >= 2 and word[0] == word[-1])

Answer 4

两者都相当不错。

有一些细微的差异：

列表理解返回您传递给 len 的另一个列表。第一个解决方案避免创建另一个列表。

Answer 5

您可能需要考虑的其他一些变体：

首先，您可以将过滤条件分解为函数。不管怎样，这个条件都很好，但如果它变得更复杂，我肯定会这样做：

def check(word):
    return len(word) >= 2 and word[0] == word[-1]
sum(1 for word in words if check(word))

接下来，如果生成一个列表（如原始列表理解中那样）是可以接受的，那么你可以这样做：

len(filter(check, words))

有 itertools.ifilter，但是如果你使用它，你需要再次使用 sum 表达式，所以它最终不会变得更清晰。

sum 技巧经常出现，以至于我很惊讶没有标准库调用来计算迭代器中的项目数（如果有，我还没有找到它）。或者，如果迭代器没有 __len__ ，那么 len 会消耗并计算迭代器中的条目数，这是有意义的，但事实并非如此。