从其他列表中查找职位并将其应用到其他列表中

时间:2018-10-26 07:17:51

标签: python list

我有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和值在运行时会有所不同。

2 个答案:

答案 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]]