代码之家  ›  专栏  ›  技术社区  ›  kiddo

重新构造嵌套python列表

  •  3
  • kiddo  · 技术社区  · 6 年前

    假设我们有一个包含3个级别(D1=2,D2=3,M=2)的嵌套列表,如下所示:

    l = [[[1,a],[2,b],[3,c]], [[4,d],[5,e],[6,f]]]
    

    我试图找出是否有任何python方法可以构建两个新的嵌套列表,其中包含两个级别(D1=2,D2=3),例如:

    l1 = [[1,2,3], [4,5,6]]
    l2 = [[a,b,c], [d,e,f]]
    

    然后我们可能有N个级别,最深级别的列表包含M个元素(D1、D2、D3、…、D\u N-1、M),目标始终是构建具有N-1级别的嵌套M列表。

    换句话说,我们需要通过拆分较低级别来保留层次结构。

    其他示例:

    l = [[[[13076, 0, 0], [806, 0, 0]], [[13076, 0, 0], [2, 0, 0]]]], [[[[2066, 0, 0], [8, 0, 0]], [[42, 0, 0], [4147, 0, 0]]]]
    
    l1 = [[[13076,806], [13076,2]], [[2066,8],[42,4147]]]
    l2 = [[[0,0], [0,0]], [[0,0], [0,0]]]
    l3 = [[[0,0], [0,0]], [[0,0], [0,0]]]
    
    4 回复  |  直到 6 年前
        1
  •  2
  •   jpp    6 年前

    numpy 语法便于执行此任务:

    import numpy as np
    
    l = [[[1, 'a'], [2, 'b'], [3, 'c']],
         [[4, 'd'], [5, 'e'], [6, 'f']]]
    
    a = np.array(l)
    
    l1 = a[:, :, 0].astype(int).tolist()
    
    # [[1, 2, 3], [4, 5, 6]]
    
    l2 = a[:, :, 1].tolist()
    
    # [['a', 'b', 'c'], ['d', 'e', 'f']]
    
        2
  •  2
  •   Dadep    6 年前

    您只需执行以下操作:

    l1=[[i[0] for i in j] for j in l]
    l2=[[i[1] for i in j] for j in l]
    
        3
  •  0
  •   Ajax1234    6 年前

    对于任意深度,可以使用递归:

    def transpose(s):
      return list(map(list, zip(*[list(map(list, zip(*i))) if all(all(not isinstance(c, list) for c in h) for h in i) else transpose(i) for i in s])))
    
    l = [[[[13076, 0, 0], [806, 0, 0]], [[13076, 0, 0], [2, 0, 0]]]], [[[[2066, 0, 0], [8, 0, 0]], [[42, 0, 0], [4147, 0, 0]]]]
    l2 = [[[1,'a'],[2,'b'],[3,'c']], [[4,'d'],[5,'e'],[6,'f']]] 
    new_data1 = {'l{}'.format(i):a for i, a in enumerate(transpose(l), start=1)}
    new_data2 = {'l{}'.format(i):a for i, a in enumerate(transpose(l2), start=1)}
    

    输出:

    {'l2': [[[[0, 0], [0, 0]]], [[[0, 0], [0, 0]]]], 'l3': [[[[0, 0], [0, 0]]], [[[0, 0], [0, 0]]]], 'l1': [[[[13076, 806], [13076, 2]]], [[[2066, 8], [42, 4147]]]]}
    
    {'l2': [['a', 'b', 'c'], ['d', 'e', 'f']], 'l1': [[1, 2, 3], [4, 5, 6]]}
    
        4
  •  0
  •   SolomonRoberts    6 年前

    在地图中使用地图

    l = [[[1,"a"],[2,"b"],[3,"c"]], [[4,"d"],[5,"e"],[6,"f"]]]
    l1 = list(map(lambda x: list(map(lambda y: y[0], x)), l))
    l2 = list(map(lambda x: list(map(lambda y: y[1], x)), l))