假如在python里有一个字典,它每个key的值都是一个列表,现在需要搜索一个字符串,判断它是否在这这个字典里,如果在,返回它的key,如果不在返回一个空,这个功能我用循环也可以做出来,但是请问有什么好的算法吗?
found = 0
for k in mydick:
if str in mydick[k]:
found = 1
print k
if not found:
print 'Noting found'
这个循环是必须的吧
好的算法不知道,但是 if str in mydick[k]:
, 就可以跳出循环了。
采用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能不能做到。更简便的方法
在列表里面执行查找操作,时间复杂度是 log(n)
;在字典中执行查找操作是 log(1)
;
看你这个功能的使用次数,如果使用次数很少,那么用其他人给的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'}
keys = [k for k, v in d.iteritems() if xxx in v]
正如二楼所说。