我有以下2D列表
list = [[1,1,a],[2,2,b],[3,3,c]]
我希望将这个2D列表转换为一个2D列表和一个数组
sublist = [[1,1],[2,2],[3,3]]
subarray = [a,b,c]
有没有方便的方法在python中做到这一点。我是python的新手,所以我不知道是否有任何语法可以执行上述计算。
答案 0 :(得分:2)
试试这个:
lst = [[1,1,a],[2,2,b],[1,3,c]]
sublist = list(map(lambda x: [x[0],x[1]], lst))
subarray = list(map(lambda x: x[2], lst))
答案 1 :(得分:1)
列表理解是一个很好的,干净的解决方案:
sublist = [[a[0], a[1]] for a in list]
subarray = [a[2] for a in list]