从字典键和值中过滤数据集

Question

假设我有一个pd.DataFrame：

df = pd.DataFrame({'Press':['A', 'B', 'A', 'B', 'A'],
                   'Model':[1, 2 ,2 , 2, 1],
                   'Count':[1, 1 ,1 , 1, 1]})

排序后，我只想保持数据匹配somedict = {'A':2, 'B':2}。（键=按，值=模型）

有一个简单的方法来只保留索引[1,2,3]吗？

Answer 1

在这里您使用 merge

s=pd.Series(somedict).to_frame('Model').rename_axis('Press').reset_index()
yourdf=df.merge(s)
Out[231]: 
  Press  Model  Count
0     B      2      1
1     B      2      1
2     A      2      1