Question

我有一个python字典，我已经构建了它的键是一个元组，值在列表中。元组包含整数和字符串。这里最好的描述是一个简单的例子：

>>> x = {(1, 'test'): ['marry', 'tom']}

我想构建一个逻辑，如果字典键中元组的第一个元素存在我的逻辑：

所以我尝试了以下内容：

>>> if 1 in x:
...     print x
...

它不起作用。

但后来我做了以下测试，如果在函数工作时键不是一个元组，它确实：

>>> y = {}
>>> y[1] = []
>>> y[1].append("tom")
>>> if 1 in y:
...     print y
... 
{1: ['tom']}

如果密钥恰好是元组的第一个元素，如何检查密钥是否存在于字典中。

Answer 1

您需要测试完整密钥：

(1, 'test') in x

元组键并不特殊，使用元组作为键并不意味着元组中的两个元素都成为单独的键。

如果您需要1和'test'作为键，则需要将它们分别添加为键，或者单独测试字典中的每个键。

您可以继承dict类型并对其进行扩展以支持您的解释：

class tupledict(dict):
    def __contains__(self, key):
        if super(tupledict, self).__contains__(key):
            return True
        return any(key in k for k in self)

演示：

>>> x = tupledict({(1, 'test'): ['marry', 'tom']})
>>> 1 in x
True
>>> (1, 'test') in x
True

Answer 2

如果您只知道密钥的第一部分，则可以查看密钥的any是否包含密钥：

if any(k[0] == 1 for k in x):

或者，获取匹配键列表：

keys = [k for k in x if k[0] == 1]

Python密钥存在：Key是整数和字符串的元组

2 个答案: