列表和数据框的交集，保留列表的副本但显示数据框中列的值

Question

找到这个 link 和我的工作有些相似。

假设我有：

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

使用上面 link 中的建议，我得到了这个：

df = pd.DataFrame.from_dict(y)
items = set(df['words'])

found = [i for i in x if i in items] 
print(found)

结果是： ['the', 'the', 'and', 'heart', 'heart']

想要能够得到word对应的值，卡住了。我想要的结果是这样的：

[3,3,4,1,1]

关于如何实现这一点有什么想法吗？将不胜感激。

Answer 1

您不需要 pandas。首先修改你的字典以将单词作为键，然后使用理解：

y2 = dict(zip(*y.values()))
[y2[i] for i in x if i in y2]

输出：[3,3,4,1,1]

pandas 中的等价物（效率低得多）是：

s = df.set_index('words')['values']
pd.Series(x).map(s).dropna()

输出：

0    3.0
1    3.0
2    4.0
5    1.0
6    1.0
dtype: float64

Intersection of list and dataframe, keeping duplicates of list but showing the values of a column in a dataframe