如何在字典中找到列表的组合?

时间:2013-07-06 22:59:17

标签: python dictionary tuples permutation

我试图在另一个带有元组的词典中找到列表的任何排列。

例如,在Dictionary中找到[1,2,3]的任意组合的最佳方法是什么,格式如下:{(1,3,2):'text',(3,1,2):'text'}

符合[1,2,3]条件的唯一匹配是(1,2,3),(1,3,2),(2,1,3),(2,3,1),(3,2,1),(3,1,2)

不符合条件的匹配包括不包含所有项目的列表(例如:(1,2)(2)),以及包含额外项目的匹配项(例如:{{1 }或(1,2,3,4))。

1 个答案:

答案 0 :(得分:2)

使用itertools.permutations()生成列表的所有排列:

from itertools import permutations

if any(tuple(perm) in yourdictionary for perm in permutations(yourlist)):
    # match found

但你真的想重新考虑你的数据结构。如果您改为创建了密钥frozenset()对象,则只需测试:

if frozenset(yourlist) in yourdictionary:
    # match found

这将是 lot 更快。

演示:

>>> from itertools import permutations
>>> yourdictionary = {(1,3,2):'text',(3,1,2):'text'}
>>> yourlist = [1, 2, 3]
>>> print any(tuple(perm) in yourdictionary for perm in permutations(yourlist))
True
>>> yourdictionary = {frozenset([1, 2, 3]): 'text', frozenset([4, 5, 6]): 'othertext'}
>>> frozenset(yourlist) in yourdictionary
True
>>> frozenset([2, 3]) in yourdictionary
False