pythonic way to iterate over part of a list

前端 未结 7 1851
日久生厌
日久生厌 2020-12-08 08:55

I want to iterate over everything in a list except the first few elements, e.g.:

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

This is concise, but

7条回答
  •  陌清茗
    陌清茗 (楼主)
    2020-12-08 09:54

    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
    

    Now you can for i in skip_heading( lines, 2 ): without worrying.

提交回复
热议问题