拆分嵌套列表的值

时间:2014-12-04 20:33:28

标签: python split nested-lists

如何拆分如下所示的嵌套列表,并使用"子列表"的组返回顶部列表。基于某些"断点"? 我在下面用for循环做了它,但我不确定它对于有数百或数千个子列表的列表是否足够好。 我也尝试摆弄answer to question 4322705,但我一开始并不真正理解答案。 这里的上下文是csv数据,可能有多个标题和数据集堆叠在文件中。

a = [['AAAA', 'b', 'c'], 
     ['b', 'c', 'd'],
     ['c', 'd', 'e'],
     ['d', 'e', 'f'],
     ['AAAA', '1', '2'],
     ['b', '2', '3'],
     ['c', '3', '4'],
     ['d', '4', '5']]
b = []
for i in a:
    if i[0] == 'AAAA':
        b.insert(0, [])
    b[0].append(i)

print "b = ", b

# Output:
# b = [[['AAAA', 'b', 'c'], 
#       ['b', '2', '3'], 
#       ['c', '3', '4'], 
#       ['d', '4', '5']], 
#      [['AAAA', '1', '2'], 
#       ['b', 'c', 'd'], 
#       ['c', 'd', 'e'], 
#       ['d', 'e', 'f']]]

0 个答案:

没有答案