如何在Python中创建多维列表?

2024-04-26 03:17:09 发布

您现在位置:Python中文网/ 问答频道 /正文

如何在基于某种条件遍历一维列表时创建多维列表。你知道吗

我在一个1d列表上迭代,每当我找到一个'\n'时,我应该用一个新的列表附加这样创建的列表,例如

 a = [1,2,3,4,5,'\n',6,7,8,9,0,'\n',3,45,6,7,2]

所以我希望它是

new_list = [[1,2,3,4],[6,7,8,9,0],[3,45,6,7,2]]

我该怎么做?请帮忙

def storeData(k):
    global dataList
    dlist = []
    for y in k:
        if y != '\n':
            dlist.append(y)
        else:
            break
    return dlist

这就是我尝试过的。你知道吗


Tags: in列表newforifdef条件global
3条回答

示例代码:

lst = [[]]
for x in a:
    if x != '\n':
        lst[-1].append(x)
    else:
        lst.append([])
print(lst)

输出:

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

我就是这样做的,但一定有更好的解决办法。你知道吗

x = 0
output = [[]]
for item in a:
    if item != "\n":
        output[x].append(item)
    else:
        x += 1
        output.append([])

print(output)

使用itertools.groupby可以完成这项工作(按分组而不是按作为换行符):

import itertools

a = [1,2,3,4,5,'\n',6,7,8,9,0,'\n',3,45,6,7,2]

new_list = [list(x) for k,x in itertools.groupby(a,key=lambda x : x!='\n') if k]

print(new_list)

我们比较关键真值以过滤出\n

结果:

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

相关问题 更多 >

    热门问题