如何从仅部分知道密钥的python dict中检索？

Question

我有一个dict，它具有我不知道其确切值的字符串类型键（因为它们是在其他地方动态生成的）。但是，我知道我想要的键包含一个特定的子字符串，并且带有该子字符串的单个键肯定在字典中。

检索此键的值的最佳或“最 Pythonic”的方法是什么？

我想到了两种策略，但都让我感到厌烦：

for k,v in some_dict.items():
    if 'substring' in k:
        value = v
        break

-- 或者--

value = [v for (k,v) in some_dict.items() if 'substring' in k][0]

第一种方法体积庞大且有些丑陋，而第二种方法更简洁，但是索引到列表理解的额外步骤（[0]）让我很恼火。有没有更好的方式来表达第二个版本，或者更简洁的方式来编写第一个版本？

Answer 1

我更喜欢第一个版本，尽管我会使用some_dict.iteritems()（如果您使用的是 Python 2），因为这样您就不必事先构建所有项目的完整列表。相反，您遍历 dict 并在完成后立即中断。

在 Python 3 上，some_dict.items(2) 已经生成字典视图，因此这已经是一个合适的迭代器。

Answer 2

可以选择使用第一个版本的性能属性编写第二个版本。

使用 generator expression 代替列表解析：

value = next(v for (k,v) in some_dict.iteritems() if 'substring' in k)

括号内的表达式将返回一个迭代器，然后您将要求提供下一个，即第一个元素。不处理其他元素。

Answer 3

这个怎么样：

value = (v for (k,v) in some_dict.iteritems() if 'substring' in k).next()

当它找到第一个匹配时会立即停止。

但它仍然具有 O(n) 复杂度，其中 n 是键值对的数量。您需要后缀列表或后缀树之类的东西来加快搜索速度。

Answer 4

class MyDict(dict):
    def __init__(self, *kwargs):
        dict.__init__(self, *kwargs)

    def __getitem__(self,x):
        return next(v for (k,v) in self.iteritems() if x in k)



# Defining several dicos ----------------------------------------------------    
some_dict = {'abc4589':4578,'abc7812':798,'kjuy45763':1002}

another_dict = {'boumboum14':'WSZE x478',
                'tagada4783':'ocean11',
                'maracuna102455':None}

still_another = {12:'jfg',45:'klsjgf'}



# Selecting the dicos whose __getitem__ method will be changed -------------       
name,obj = None,None
selected_dicos = [ (name,obj) for (name,obj) in globals().iteritems()
                   if type(obj)==dict
                   and all(type(x)==str for x in obj.iterkeys())]

print 'names of selected_dicos ==',[ name for (name,obj) in selected_dicos] 



# Transforming the selected dicos in instances of class MyDict -----------
for k,v in selected_dicos:
    globals()[k] = MyDict(v)



# Exemple of getting a value ---------------------------------------------      
print "some_dict['7812'] ==",some_dict['7812']

结果

names of selected_dicos == ['another_dict', 'some_dict']
some_dict['7812'] == 798

Answer 5

如果有很多键，但字符串很容易从子字符串重构，那么它可以更快地重构它。例如通常您知道密钥的开始，但不知道附加的日期戳。 （例如，您可能只需要尝试 365 个日期，而不是遍历数百万个键）。 不太可能是这种情况，但我想我还是会建议的。 例如

>>> names={'bob_k':32,'james_r':443,'sarah_p':12}
>>> firstname='james' #you know the substring james because you have a list of firstnames
>>> for c in "abcdefghijklmnopqrstuvwxyz":
...     name="%s_%s"%(firstname,c)
...     if name in names:
...             print name
... 
james_r