我有一个项目列表(HTML表行,用Beautiful Soup提取),我需要遍历列表并获得每个循环运行的偶数和奇数元素(我的意思是索引)。 我的代码如下所示:
for top, bottom in izip(table[::2], table[1::2]):
#do something with top
#do something else with bottom
如何让这段代码变得不那么难看?或者也许是这样做的好方法?
编辑:
table[1::2], table[::2] => table[::2], table[1::2]
答案 0 :(得分:5)
izip
是一个不错的选择,但是由于你对它不满意,所以有一些选择:
>>> def chunker(seq, size):
... return (tuple(seq[pos:pos+size]) for pos in xrange(0, len(seq), size))
...
>>> x = range(11)
>>> x
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> chunker(x, 2)
<generator object <genexpr> at 0x00B44328>
>>> list(chunker(x, 2))
[(0, 1), (2, 3), (4, 5), (6, 7), (8, 9), (10,)]
>>> list(izip(x[1::2], x[::2]))
[(1, 0), (3, 2), (5, 4), (7, 6), (9, 8)]
正如您所看到的,这样做的好处是可以正确处理不均匀的元素,这对您来说可能不重要。还有来自itertools documentation itself:
的这个食谱>>> def grouper(n, iterable, fillvalue=None):
... "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
... args = [iter(iterable)] * n
... return izip_longest(fillvalue=fillvalue, *args)
...
>>>
>>> from itertools import izip_longest
>>> list(grouper(2, x))
[(0, 1), (2, 3), (4, 5), (6, 7), (8, 9), (10, None)]
答案 1 :(得分:4)
尝试:
def alternate(i):
i = iter(i)
while True:
yield(i.next(), i.next())
>>> list(alternate(range(10)))
[(0, 1), (2, 3), (4, 5), (6, 7), (8, 9)]
此解决方案适用于任何序列,而不仅仅是列表,并且不会复制序列(如果您只需要长序列的前几个元素,它将更有效。)
答案 2 :(得分:0)
看起来不错。我唯一的建议是将它包装在一个函数或方法中。这样,您可以为其命名(evenOddIter()
),使其更具可读性。