我有一个对象列表,每个对象都包含几个类变量。我对其中的几个类变量感兴趣,我想创建两个单独的列表,每个列表都包含来自该对象列表的不同类变量。
对象类看起来像这样:
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]
它有效,但我想知道是否有一种通过列表理解来实现这一目标的一步法?或者你推荐的其他方式?提前致谢!
答案 0 :(得分:5)
列表推导式的定义就是生成一个列表对象。
不要在这里使用列表推导式。只需使用一个普通的循环:
list_of_array1 = []
list_of_array2 = []
for x in list:
list_of_array1.append(x.array1)
list_of_array2.append(x.array2)
这让您只需要执行一个循环。
答案 1 :(得分:2)
您可以使用 zip
函数:
L1, L2 = zip(*[(x.array1,x.array2) for x in list])
由你来改变理解...
答案 2 :(得分:2)
要仅迭代 list
一次,但创建两个结果列表,请使用普通的 for
循环:
list_of_array1 = []
list_of_array2 = []
for x in list:
list_of_array1.append(x.array1)
list_of_array2.append(x.array2)
但是您使用两个列表推导式的原始解决方案也很好。性能差异可能可以忽略不计。