列表与DataFrame求交集保留列表重复值并获取对应列值的方法

你可以先将y中的词和对应数值转换为映射字典，遍历x的过程中直接筛选并取值即可，无需单独先筛选匹配的词，示例代码如下：

x = ['the', 'the', 'and', 'a', 'apple', 'heart', 'heart']
y = {'words': ['the', 'belt', 'computer', 'heart','and'],'values':[3,2,1,1,4]}

# 构建词到数值的映射字典
word_value_map = dict(zip(y['words'], y['values']))
# 遍历x，匹配到的词直接取对应数值
result = [word_value_map[word] for word in x if word in word_value_map]

print(result)

运行输出就是你期望的[3, 3, 4, 1, 1]。

如果你要沿用已有的pandas写法，只需要把词集合替换为索引映射即可，修改后代码如下：

import pandas as pd

x = ['the', 'the', 'and', 'a', 'apple', 'heart', 'heart']
y = {'words': ['the', 'belt', 'computer', 'heart','and'],'values':[3,2,1,1,4]}

df = pd.DataFrame.from_dict(y)
# 构建词到数值的映射
word_value_map = df.set_index('words')['values'].to_dict()

result = [word_value_map[i] for i in x if i in word_value_map]
print(result)

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