如何将2D列表分成一个2D列表和一个数组(删除最后一列)

时间:2015-01-31 01:13:53

标签: python linear-regression

我有以下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的新手,所以我不知道是否有任何语法可以执行上述计算。

2 个答案:

答案 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]