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))。