首页 >后端开发 >Python教程 >如何在 Python 中以特定大小的块优雅地迭代列表?

如何在 Python 中以特定大小的块优雅地迭代列表?

DDD
DDD原创
2024-12-18 20:56:20934浏览

How Can I Elegantly Iterate Through a List in Python in Specific-Sized Chunks?

优雅的块:按组迭代列表

以特定大小的块迭代列表可能是 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中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn