一次使用 Python 列表理解來創建兩個列表？

Question

我有一個對象列表，每個對象都包含幾個類變量。 我對其中的幾個類變量感興趣，我想創建兩個單獨的列表，每個列表都包含來自該對象列表的不同類變量。

對象類看起來像這樣：

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]

它有效，但我想知道是否有一種通過列表理解來實現這一目標的一步方法？ 或者你推薦的其他方式？ 提前致謝！

Answer 1

列表推導式的定義就是生成一個列表對象。

不要在這里使用列表推導式。 只需使用一個普通的循環：

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

這讓您只需要執行一個循環。

Answer 2

您可以使用zip功能：

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

由你來改變理解...

Answer 3

要僅迭代list一次，但創建兩個結果列表，請使用普通的for循環：

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

但是您使用兩個列表推導式的原始解決方案也很好。 性能差異可能可以忽略不計。