python - 在 Python 中解压给定数量的项目？

Question

有没有什么方法可以使用“splat”运算符(例如 a, *rest = somelist)来消耗一定数量的项目？

用例:我想将一些输入拆分为一个数字、一个列表列表、另一个数字和另一个列表列表。

我的输入是这样的:

5
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
5
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16

我想要这样的名称 first_num、first_arrangement、second_num、second_arrangement:

first_num == 5
first_arrangement == [[1, 2, 3, 4], [5, 6, 7, 8], ...]

等等。

要做到这一点，能够从生成行的可迭代对象中消耗一定数量的项目会很有用。像这样的东西作为中间步骤是理想的:
first_num, *[4]first_arrangement, second_num, *[4]second_arrangement = 行


解决这个问题的正常/规范/Python 方法是什么？

Answer 1

我认为执行此操作的规范的 Pythonic 方法是将责任放在您正在迭代的生成器上。我会像这样定义一个生成器函数:

import itertools

def generate_arrangements(iterable, size=4):
    iterator = iter(iterable)
    while True:
        yield next(iterator)
        yield list(list(row) for row in itertools.islice(iterator, size))

假设您的数据在这样的列表中:

data = [
    5,
    [1, 2, 3, 4],
    [5, 6, 7, 8],
    [9, 10, 11, 12],
    [13, 14, 15, 16],
    5,
    [1, 2, 3, 4],
    [5, 6, 7, 8],
    [9, 10, 11, 12],
    [13, 14, 15, 16]
]

然后写:

first_num, first_arr, second_num, second_arr = generate_arrangements(data)

给你你想要的输出:

>>> first_num
5
>>> first_arr
[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]

你也可以同时产生数字和排列，让用户做一些额外的拆包，这可能是一种更自然的方法:

import itertools

def generate_arrangements(iterable):
    iterator = iter(iterable)
    while True:
        number =  next(iterator)
        arrangement = list(list(row) for row in itertools.islice(iterator, 4))
        yield number, arrangement

(first_num, first_arr), (second_num, second_arr) = generate_arrangements(data)

正如@JoranBeasley 在评论中所写，这种形式使得在 for 循环中使用元组解包变得容易，如下所示:

for num,arr in generate_arrangements(data):
    print(num) 
    print(arr)