在python里判断一个值是否是一个字典里如果有返回key有什么好的算法?

假如在python里有一个字典,它每个key的值都是一个列表,现在需要搜索一个字符串,判断它是否在这这个字典里,如果在,返回它的key,如果不在返回一个空,这个功能我用循环也可以做出来,但是请问有什么好的算法吗?

found = 0
for k in mydick:    
    if str in mydick[k]:
        found = 1
        print k        
if not found:
    print 'Noting found'
阅读 25k
5 个回答

采用list comprehension就可以啦,示范如下

# 定义一个dictionary s, 查询包含5的list并且返回相应的key
s = dict(a=[1,2,3,4], b=[2,3,4,5], c=[3,5,7,8], d=[1,2,5,9])
# 采用list comprehension, python自动的就能很好的处理运行效率
result = [k for k, v in s.iteritems() if 5 in v]
# print result 就得到包含所有key的一个list
# ['c', 'b', 'd'],判断返回的list是否为空(e.g. if len(result) == 0)就知道是否有值了。

期待高手给更好的答案,不知道lambda能不能做到。更简便的方法

看你这个功能的使用次数,如果使用次数很少,那么用其他人给的O(n^2)的方法也无所谓。

如果这个功能被调用次数很多,那么我推荐翻转字典,然后查询翻转后的字典。

from collections import defaultdict

def inverse(d):
    inv_d = defaultdict(set)

    def inverse_update(k, v):
        inv_d[v].add(k)

    for k, vs in d.items():
        for v in vs:
            inverse_update(k, v)

    return inverse_update, inv_d


inv_update_d, inv_d = inverse({'a': [1, 2, 3],
                               'b': [2, 3, 4],
                               'c': [4, 5, 6]})
# 如果你更新了原字典d,那么记得用inv_update_d更新翻转后的字典inv_d

print(inv_d.get(5, None)) # => {'c'}
print(inv_d.get(10, None)) # => None

inv_update_d('d', 10)
print(inv_d.get(10, None)) # => {'d'}

这个循环是必须的吧
好的算法不知道,但是 if str in mydick[k]:, 就可以跳出循环了。

新手上路,请多包涵

keys = [k for k, v in d.iteritems() if xxx in v]
正如二楼所说。

在列表里面执行查找操作,时间复杂度是 log(n);在字典中执行查找操作是 log(1)

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题
logo
101 新手上路
子站问答
访问
宣传栏