我有3个列表,例如
list1 = [30,14,42,17,18]
list2 = [17,18,6,7,30,9,5,14,3,42,33]
list3 = [[-25,54,789,0.56],[-5,4,9,0.5],[-5,4,0.009,0.8],[-5,456,7,56],
[-5,4,9,0.6],[-2,4,9,-0.578],[-2,0.5,-7,-0.6],[-25,-50,78,0.1],[-4,-58,79,0.6],
[-2,5,7,4],[-5,54,43,0.45]]
list2和list3的长度相同。
我的问题是取list1
中的每个值,在list2
中找到其位置,从同一位置的list3
中提取值。这些列表是在运行时生成的,即ID和值在运行时会有所不同。
答案 0 :(得分:3)
您可以对enumerate
使用dict理解来为dict2
创建数字到索引的映射,然后使用该映射将list1
中的数字映射到相应的子列表在list3
中的映射索引处:
mapping = {n: i for i, n in enumerate(list2)}
print([list3[mapping[n]] for n in list1])
这将输出:
[[-5, 4, 9, 0.6], [-25, -50, 78, 0.1], [-2, 5, 7, 4], [-25, 54, 789, 0.56], [-5, 4, 9, 0.5]]
请注意,首先创建映射将问题的平均时间复杂度降低为 O(n)。
答案 1 :(得分:2)
使用列表理解,
In [67]: print [list3[j] for j in [list2.index(i) for i in list1]]
Out[67]:
[[-5, 4, 9, 0.6],
[-25, -50, 78, 0.1],
[-2, 5, 7, 4],
[-25, 54, 789, 0.56],
[-5, 4, 9, 0.5]]
或
In [69]: print [list3[list2.index(i)] for i in list1]
Out[69]:
[[-5, 4, 9, 0.6],
[-25, -50, 78, 0.1],
[-2, 5, 7, 4],
[-25, 54, 789, 0.56],
[-5, 4, 9, 0.5]]