一次使用 Python 列表理解来创建两个列表？

问题描述

我有一个对象列表，每个对象都包含几个类变量。我对其中的几个类变量感兴趣，我想创建两个单独的列表，每个列表都包含来自该对象列表的不同类变量。

对象类看起来像这样：

class Pie: 
     def __init__(self,array1,array2,timestamp,latitude,longitude):
          self.array1 = array1
          self.array2 = array2
          self.timestamp = timestamp
          self.latitude = latitude
          self.longitude = longitude

我有一个包含一堆这些对象的列表：

list = [pie_instance1,pie_instance2,pie_instance3,pie_instance4,pie_instance5]

我想创建一个包含 array1 的列表，以及另一个包含 array2 的列表。

我一开始只使用了两次列表理解：

list_of_array1 = [x.array1 for x in list]
list_of_array2 = [x.array2 for x in list]

但是遍历整个列表两次似乎效率不高（我实际代码中的列表非常大）。

所以我尝试了这个：

temp = np.array([[x.array1,x.array1] for x in list])
list_of_array1 = temp[:,0]
list_of_array2 = temp[:,1]

它有效，但我想知道是否有一种通过列表理解来实现这一目标的一步法？或者你推荐的其他方式？提前致谢！

解决方法

列表推导式的定义就是生成一个列表对象。

不要在这里使用列表推导式。只需使用一个普通的循环：

list_of_array1 = []
list_of_array2 = []
for x in list:
    list_of_array1.append(x.array1)
    list_of_array2.append(x.array2)

这让您只需要执行一个循环。

,

您可以使用 zip 函数：

L1,L2 = zip(*[(x.array1,x.array2) for x in list])

由你来改变理解...

,

要仅迭代 list 一次，但创建两个结果列表，请使用普通的 for 循环：

list_of_array1 = []
list_of_array2 = []
for x in list:
    list_of_array1.append(x.array1)
    list_of_array2.append(x.array2)

但是您使用两个列表推导式的原始解决方案也很好。性能差异可能可以忽略不计。