Question

我有这个嵌套列表：

list_1 = [[1,2,3], [1,2,3,4,5,6], [1,2,3,4,5,6,7,8,9]]

子列表元素的数量始终为3。我希望每个子列表中有3个元素。所需的输出：

list_1 = [[1,2,3], [1,2,3], [4,5,6],[1,2,3], [4,5,6], [7,8,9]]

我可以实现这一点，但首先我必须将列表展平，然后创建嵌套列表。我的代码：

list_1 = [values for sub_list in lists_1 for values in sub_list]  # flatten it first

list_1 = [list_1[i:i+3] for i in range(0, len(list_1), 3)]

是否可以跳过展平步骤并获得所需结果？

Answer 1

您可以使用嵌套列表理解：

list_1 = [[1,2,3], [1,2,3,4,5,6], [1,2,3,4,5,6,7,8,9]]
result = [i[j:j+3] for i in list_1 for j in range(0, len(i), 3)]

输出：

[[1, 2, 3], [1, 2, 3], [4, 5, 6], [1, 2, 3], [4, 5, 6], [7, 8, 9]]

Answer 2

以下是使用嵌套列表推导的方法：

list_1 = [[1,2,3],[1,2,3,4,5,6],[1,2,3,4,5,6,7,8,9]]

list_1 = [a for b in list_1 for a in b]
list_1 = [list_1[i:i+3] for i in range(0,len(list_1),3)]

print(list_1)

输出：

[[1, 2, 3], [1, 2, 3], [4, 5, 6], [1, 2, 3], [4, 5, 6], [7, 8, 9]]

Answer 3

要投入两分钱，您可以使用两个生成器函数，一个函数使列表变平（带有任意嵌套的列表），而一个函数产生成对的n个值：

def recursive_yield(my_list):
    for item in my_list:
        if isinstance(item, list):
            yield from recursive_yield(item)
        else:
            yield item

def taken(gen, number = 3):
    buffer = []
    for item in gen:
        if len(buffer) < number:
            buffer.append(item)
        else:
            yield buffer
            buffer = []
            buffer.append(item)
    if buffer:
        yield buffer
    
result = [x for x in taken(recursive_yield(list_1))]

以下是输入/输出的一些示例：

list_1 = [[1,2,3], [1,2,3,4,5,6], [1,2,3,4,5,6,7,8,9]]
# -> [[1, 2, 3], [1, 2, 3], [4, 5, 6], [1, 2, 3], [4, 5, 6], [7, 8, 9]]

list_1 = [1,2,3,4,5,6]
# -> [[1, 2, 3], [4, 5, 6]]

list_1 = [1,2,[[1,2,4,5], [[[[1,10,9]]]]]]
# -> number = 5
# -> [[1, 2, 1, 2, 4], [5, 1, 10, 9]]

因此，该解决方案比单独切片要灵活得多。

在python的嵌套列表中创建嵌套列表

3 个答案: