优雅的块:按组迭代列表
以特定大小的块迭代列表可能是 Python 编程中的常见任务。如果列表包含要在预定义组中处理的元素,典型的方法可能类似于提供的代码:
for i in range(0, len(ints), 4): # Placeholder operation for illustrative purposes foo += ints[i] * ints[i + 1] + ints[i + 2] * ints[i + 3]
虽然这种方法可以满足功能要求,但它可能会让人感觉相当笨重且不符合 Python 风格。本文深入研究了使用自定义生成器函数的更优雅和惯用的解决方案:
def chunker(seq, size): return (seq[pos:pos + size] for pos in range(0, len(seq), size))
这个生成器函数,chunker,优雅地将输入序列分割成指定大小的块,提供了一种简化的迭代方式他们。 chunker 中的生成器表达式生成每个块,从序列的开头开始并按块大小前进,直到到达或超过序列的末尾。
以下示例演示了 chunker 函数的多功能性:
text = "I am a very, very helpful text" for group in chunker(text, 7): print(repr(group),)
输出:
'I am a ' 'very, v' 'ery hel' 'pful te' 'xt'
此示例将文本字符串分成 7 个字符的组 每个。同样,它可以用于任何类型的序列,包括动物列表甚至任意数据结构。
在 Python 3.12 及更高版本中,可以使用 itertools.tee() 函数进一步简化 chunker 函数:
from itertools import tee def chunker(seq, size): a, b = tee(seq, 2) return (list(islice(a, i, None, size)) for i in range(size))
该函数的这种变体避免了创建中间列表并直接对输入序列进行操作,突出了 Python 内置函数在处理可迭代。
以上是如何在 Python 中以特定大小的块优雅地迭代列表?的详细内容。更多信息请关注PHP中文网其他相关文章!