如果你想将一个列表分成多个切片,你可以使用这个技巧:
list_of_slices = zip(*(iter(the_list),) * slice_size)
例如
>>> zip(*(iter(range(10)),) * 3)
[(0, 1, 2), (3, 4, 5), (6, 7, 8)]
如果项目的数量不能被切片大小整除,并且你想用 None 填充列表,你可以这样做:
>>> map(None, *(iter(range(10)),) * 3)
[(0, 1, 2), (3, 4, 5), (6, 7, 8), (9, None, None)]
这是一个肮脏的小把戏
好的,我会解释它是如何工作的。解释起来会很棘手,但我会尽力而为。
先说一点背景:
在 Python 中,您可以将列表乘以这样的数字:
[1, 2, 3] * 3 -> [1, 2, 3, 1, 2, 3, 1, 2, 3]
([1, 2, 3],) * 3 -> ([1, 2, 3], [1, 2, 3], [1, 2, 3])
一个迭代器对象可以像这样被消费一次:
>>> l=iter([1, 2, 3])
>>> l.next()
1
>>> l.next()
2
>>> l.next()
3
zip函数返回一个元组列表,其中第i 个元组包含来自每个参数序列或迭代的第 i 个元素。例如:
zip([1, 2, 3], [20, 30, 40]) -> [(1, 20), (2, 30), (3, 40)]
zip(*[(1, 20), (2, 30), (3, 40)]) -> [[1, 2, 3], [20, 30, 40]]
zip 前面的 * 用于解压缩参数。您可以在此处找到更多详细信息。所以
zip(*[(1, 20), (2, 30), (3, 40)])
实际上相当于
zip((1, 20), (2, 30), (3, 40))
但适用于可变数量的参数
现在回到诀窍:
list_of_slices = zip(*(iter(the_list),) * slice_size)
iter(the_list)
-> 将列表转换为迭代器
(iter(the_list),) * N
-> 将生成一个对 the_list 迭代器的 N 引用。
zip(*(iter(the_list),) * N)
-> 会将这些迭代器列表提供给 zip。这反过来又会将它们分组为 N 个大小的元组。但是由于所有 N 项实际上都是对同一个迭代器iter(the_list)
的引用,因此结果将next()
在原始迭代器上重复调用
我希望能解释它。我建议您使用更易于理解的解决方案。我只是想提及这个技巧,因为我喜欢它。