迭代列表的一部分的pythonic方法答案

【问题标题】：pythonic way to iterate over part of a list迭代列表的一部分的pythonic方法
【发布时间】：2012-01-30 01:45:36
【问题描述】：

我想遍历列表中除前几个元素之外的所有内容，例如：

for line in lines[2:]:
    foo(line)

这很简洁，但会复制整个列表，这是不必要的。我可以这样做：

del lines[0:2]
for line in lines:
    foo(line)

但这会修改列表，这并不总是好的。

我可以这样做：

for i in xrange(2, len(lines)):
    line = lines[i]
    foo(line)

但是，这太恶心了。

可能会更好：

for i,line in enumerate(lines):
    if i < 2: continue
    foo(line)

但它不像第一个例子那么明显。

那么：有什么方法可以做到与第一个示例一样明显，但又不会不必要地复制列表？

【问题讨论】：

我觉得for i in xrange(2, len(lines)): 很好。
我知道这并不能回答您的问题，但是对于仅 300 行文本，我完全不用担心复制。请记住，只有引用被复制，而不是字符串数据本身......
就我个人而言，我喜欢你提出的第 4 个例子。它的效率不亚于islice 方法，而且在我看来，其他任何查看您的代码的人都会更容易阅读和理解。
我也认为for i in xrange(2, len(lines)) 很好，比以下任何选择都好。为什么你认为它“恶心”？
@BlueRaja-DannyPflughoeft：因为 Python 式的迭代方式是使用 for-each 循环：for x in stuff:。你不这样做for i in xrange(len(stuff)): x=stuff[i].. 那么如果你只是想做同样的迭代但略有不同，你为什么要这样做呢？

标签： python list iteration

【解决方案1】：

你可以试试itertools.islice(iterable[, start], stop[, step]):

import itertools
for line in itertools.islice(list , start, stop):
     foo(line)

【讨论】：

@NullUserException 干杯！本来要补充的，但我的比赛打了个嗝
很棒的是：for line in lazy(list)[:2]: foo(line)，其中lazy 包装了可迭代对象，如果将其切片，则调用islice，等等。这样它看起来就像一个切片（它应该）。
@Claudiu 因为切片应该是与切片相同类型的对象，这意味着lazy() 创建了一个迭代器。是discussed on python mailing list，Guido 反对这个想法。这不会停止独立实现，但您不会很快在标准库中看到它。
警告：与list1[start:stop] 相比，可能非常效率低下。例如，islice(list1, 100000, 100100) 将首先迭代前 99,999 个值（只是不会返回它们）。用 cProfile 自己试试；它也在文档中。
@Jeff Upvoted 在调用 next start - 1 时间超重列表切片内存分配的情况下是正确的。 OP 在他的问题中特别提到列表切片是一种不受欢迎的解决方案。

【解决方案2】：

在大多数情况下，最初的解决方案是合适的。

for line in lines[2:]:
    foo(line)

虽然这确实复制了列表，但它只是一个浅拷贝，而且速度非常快。在您分析代码并发现这是一个瓶颈之前，不要担心优化。

【讨论】：

...但是如果您知道您将使用包含数千或数百万个元素的列表，请不要嘲笑itertools.islice 这样的礼物。在这种情况下，值得一试。
@ChrisMorgan：“在大多数情况下”这句话并不是冷嘲热讽。
即使它没有针对大型列表进行优化，这段代码也具有小型列表的可读性。

【解决方案3】：

虽然itertools.islice 似乎是解决此问题的最佳解决方案，但不知何故，对于如此简单的事情，额外的导入似乎有点过头了。

就个人而言，我发现enumerate 解决方案非常易读且简洁——尽管我更喜欢这样写：

for index, line in enumerate(lines):
    if index >= 2:
        foo(line)

【讨论】：

啊，我个人更喜欢缩进越少越好。错误检查 + continue/return/break/etc 先，然后剩下的就失败了
@Claudiu：你所说的也立即传达了意图；您无需查看 if 块的末尾以查看它之后或 else 块中是否发生任何事情。
@ChrisMorgan。多就是少。在示例中，单独的continue 会在代码中创建一个冗余分支，不必要地迫使读者解析两个代码块，而不是一个。如果循环内的代码更复杂，那么continue 语句当然可以通过减少缩进来提高可读性。但这不是该示例试图说明的内容。
@ekhumoro：确实，这取决于块的长度。我当然不会以continue 的方式做不到六行。

【解决方案4】：

您可以构建一个辅助生成器：

def rangeit(lst, rng):
  for i in rng:
    yield lst[i]

for e in rangeit(["A","B","C","D","E","F"], range(2,4)):
  print(e)

【讨论】：

不错。但它仅限于列表，不适用于任意迭代。
@S.Lott 你当然是对的。但它符合 OP 的要求，是我想到的第一件事 - 不幸的是，我不像其他人那样记住 itertools 的所有功能;-)

【解决方案5】：

for fooable in (line for i,line in enumerate(lines) if i >= 2):
    foo(fooable)

【讨论】：

【解决方案6】：

def skip_heading( iterable, items ):
    the_iter= iter( iterable ):
    for i, e in enumerate(the_iter):
        if i == items: break
    for e in the_iter:
        yield e

现在您可以for i in skip_heading( lines, 2 ): 不用担心了。

【讨论】：

这基本上是itertools.islice的特例。
@delnan：同意。它有点可扩展。有时真正的问题是头部/尾部处理，其中头部有一种处理，尾部有一种不同的处理。
这将如何工作？我只是将它复制到一个小型测试项目中，它没有打印任何东西——这也是我期望它做的。
我错过了什么吗？看起来items 在此函数中未使用。我认为您有错字/错误。
@S.Lott 我在 python 中写的代码不多，假设代码中有一个我乍一看不明白的错误不是我想做的事情；）你的修复是直截了当的，所以没有隐藏某种聪明的把戏，但谁知道呢。

【解决方案7】：

我更喜欢为此使用dropwhile。在 Haskell 和其他一些语言中使用它后感觉很自然，并且看起来相当清晰。您还可以在许多其他情况下使用它，在这些情况下，您希望在迭代开始时寻找比项目索引更复杂的“触发”条件。

from itertools import dropwhile

for item in dropwhile(lambda x: x[0] < 2, enumerate(lst)):
  # ... do something with item

【讨论】：