python - 多对一映射(创建等价类)

Question

我有一个将一个数据库转换为另一个数据库的项目。原始数据库列之一定义行的类别。此列应映射到新数据库中的新类别。

例如，假设原始类别是:parrot、spam、cheese_shop、Cleese、Gilliam、Palin

这对我来说有点冗长，我想将这些行归类为 sketch, actor - 即，将所有草图和所有 Actor 定义为两个等价类。

>>> monty={'parrot':'sketch', 'spam':'sketch', 'cheese_shop':'sketch', 
'Cleese':'actor', 'Gilliam':'actor', 'Palin':'actor'}
>>> monty
{'Gilliam': 'actor', 'Cleese': 'actor', 'parrot': 'sketch', 'spam': 'sketch', 
'Palin': 'actor', 'cheese_shop': 'sketch'}

这很尴尬——我更喜欢这样的东西:

monty={ ('parrot','spam','cheese_shop'): 'sketch', 
        ('Cleese', 'Gilliam', 'Palin') : 'actors'}

但这当然会将整个元组设置为键:

>>> monty['parrot']

Traceback (most recent call last):
  File "<pyshell#29>", line 1, in <module>
    monty['parrot']
KeyError: 'parrot'

关于如何在 Python 中创建优雅的多对一字典有什么想法吗？

Answer 1

在我看来，您有两个顾虑。首先，你最初是如何表达你的映射的，也就是说，你如何将映射输入到你的 new_mapping.py 文件中。其次，映射在重新映射过程中如何工作。这两个表示没有理由相同。

从您喜欢的映射开始:

monty = { 
    ('parrot','spam','cheese_shop'): 'sketch', 
    ('Cleese', 'Gilliam', 'Palin') : 'actors',
}

然后转换成你需要的映射:

working_monty = {}
for k, v in monty.items():
    for key in k:
        working_monty[key] = v

制作:

{'Gilliam': 'actors', 'Cleese': 'actors', 'parrot': 'sketch', 'spam': 'sketch', 'Palin': 'actors', 'cheese_shop': 'sketch'}

然后使用 working_monty 完成工作。