从 Python 上的 dict 中获取集合运算符

Question

如果我有两个或更多集合，以及一个描述必须在它们之间完成的 de 操作的字符串，例如 'and'、'or'、'xor'，它显然可以像这样完成：

if string == 'and': 
return set1.intersection(set2)
elif string == 'or'
return set1 | set2

等等。如果我想用字典做呢？ 我有这样的东西：

dictionary = {'and': set.intersection, 'or': set.union}
return set1.dictionary[string](set2)

并且还尝试过

operation = dictionary.get(string)
return set1.operation(set2)

但没有一个奏效。我怎样才能达到与通过 ifs 但使用字典获得的相同结果？

Answer 1

你可以使用set.itersection()、set.union()等作为静态方法，将多个集合传递给它们：

>>> ops = {'and': set.intersection, 'or': set.union}
>>> set1 = {1, 2, 3}
>>> set2 = {3, 4, 5}
>>> ops['and'](set1, set2)
{3}
>>> ops['or'](set1, set2)
{1, 2, 3, 4, 5}

或者，您可以将操作映射到方法名称并使用getattr()：

>>> ops = {'and': 'intersection', 'or': 'union'}
>>> getattr(set1, ops['and'])(set2)
{3}
>>> getattr(set1, ops['or'])(set2)
{1, 2, 3, 4, 5}

Answer 2

你可以试试attrgetter

from operator import attrgetter
set_methods = {"and": attrgetter("intersection"), 'or': attrgetter("union")}

这样称呼它：set_methods[method_string](set1)(set2).