python - 当字典键是列表时将字典映射到数据框

Question

我有一本字典，其中的值是列表:

dict = {1: ['a','b'], 2:['c', 'd']}

我想将字典映射到数据框的 col1 上。

col1    
 a     
 c     

如果 col1 的值是我字典的值之一，那么我想用字典键的值替换 col1 的值。

像这样，我的数据框将变成:

col1
 1
 2

提前致谢

Answer 1

我会以正确的方式转换字典:

mapping = {}
for key, values in D.items():
    for item in values:
        mapping[item] = key

然后

df['col1'] = df['col1'].map(mapping)