Python：可以在单个列表理解语句中完成

Question

我想要做的是提取列表中的第零个元素和给定二维列表的另一个列表中的第一个元素。

baseball = [[180, 78.4], 
      [215, 102.7], 
      [210, 98.5], 
      [188, 75.2]] 

x = [ a[0] for a in baseball ] 
y = [ a[1] for a in baseball ] 
print x 
print y 

这可以在单个列表理解语句中完成吗？

Answer 1

假设它的矩形（即内部列表的长度是一致的。），可以实现如下：

def transpose(matrix): 
    return [[matrix[j][i] for j in range(len(matrix))] for i in range(len(matrix[0]))] 

然后，你的问题是刚刚转（x, y = transpose(baseball)）的调用。

Answer 2

如果你不介意的元组：

baseball = [[180, 78.4], 
      [215, 102.7], 
      [210, 98.5], 
      [188, 75.2]] 


x,y = zip(*baseball) 

如果你真的想名单：

x,y = map(list,zip(*baseball)) 

如果你有在每两个以上的元素，并希望只是某些元素，如：

baseball = [[180, 1, 78.4], 
      [215, 2, 102.7], 
      [210, 3, 98.5], 
      [188, 4, 75.2]] 

from operator import itemgetter 

x, y = zip(*map(itemgetter(0, 2), baseball)) 

这会给你：

((180, 215, 210, 188), (78.4, 102.7, 98.5, 75.2))