如何获取两列表公共整数元素的索引并映射至第三列表

让我们一步步拆解你的问题，先分析现有代码的问题，再给出正确的实现：

你的代码问题分析 #

第一段代码的问题 #

p_Q = set(Q_act).intersection(A_set)
p_dur_Q_act = [i + 1 for i, x in enumerate(p_Q)]
print(p_dur_Q_act)

这段代码有三个核心问题：

1. 集合无序性：set(Q_act).intersection(A_set)得到的是无序集合，会丢失Q_act原本的元素顺序（这次碰巧结果是[1,2]，但如果Q_act是[3,2]，结果可能会乱序）。
2. 错误的索引计算：你遍历的是集合的枚举索引，而不是元素在A_set中的真实索引——i + 1只是集合元素的位置+1，不是A_set里的索引（这只是巧合和你的测试数据匹配）。
3. 未提取dur的值：这段代码只生成了索引列表，根本没有用索引去dur中取值，所以自然得不到目标的[4,5]。

第二段代码的问题 #

p_dur_Q_act = [i + 1 for i, x in enumerate(Q_act) if any(elem in x for elem in A_set)]
print(p_dur_Q_act)

这里的错误很明确：any(elem in x for elem in A_set)中，x是Q_act里的整数元素（比如2），你试图检查elem是否在整数里，但整数是不可迭代的，所以触发TypeError: argument of type 'int' is not iterable。而且逻辑上也搞反了，应该是检查x是否在A_set里，而不是反过来。

正确的实现方式 #

推荐用元素-索引映射字典的方法，既高效又能保持Q_act的顺序：

A_set = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
Q_act = [2, 3]
dur = [0, 4, 5, 2, 1, 3, 4, 8, 2, 3]

# 第一步：创建A_set元素到索引的映射字典，O(n)时间复杂度
elem_to_idx = {elem: idx for idx, elem in enumerate(A_set)}

# 第二步：遍历Q_act，提取对应dur的值，保持原顺序
p_dur_Q_act = [dur[elem_to_idx[elem]] for elem in Q_act if elem in elem_to_idx]

print(p_dur_Q_act)  # 输出: [4,5]

为什么这个方法更好？ #

• 高效：字典的查找是O(1)，比每次遍历A_set找索引的O(n)方法快得多，尤其当A_set很大时。
• 保序：严格遵循Q_act的元素顺序，不会因为集合无序而打乱。
• 容错：通过if elem in elem_to_idx过滤掉Q_act中不在A_set里的元素（如果需要保留这些元素并设置默认值，可以用dur[elem_to_idx.get(elem, -1)]或者自定义默认值）。

备选方案（适合小列表） #

如果你的列表很小，也可以直接嵌套遍历（但效率较低）：

p_dur_Q_act = [dur[idx] for elem in Q_act for idx, a_elem in enumerate(A_set) if a_elem == elem]
print(p_dur_Q_act)  # 输出: [4,5]

内容的提问来源于stack exchange，提问作者AJ Gayeta