[英]python- searching dictionary sublists; converting dictionary keys to values
说我有以下字典(我正在使用的字典更多,更大):
dict1={1:["item", "word", "thing"], 2:["word", "item"], 3:["thing", "item", "item"]}
并将字典中使用的每个单词存储在列表中:
all_words=["item", "word", "thing"]
我想通过字典子列表运行列表中的每个单词,并返回找到它们的所有子列表的键,将它们存储在元组中。 所以我想得到:
dict2={"item":(1, 2, 3), "word":(1, 2), "thing":(1, 3)}
继承人我所拥有的:
dict2={}
for word in all_words:
for key, sublist in dict2.items():
for word in sublist:
if word not in sublist:
dict2[word]=dict2[word]+key
else:
dict2[word]=key
因此,基于评论的固定程序将如下所示
>>> dict2 = {}
>>> for word in all_words:
... # Iterate over the dict1's items
... for key, sublist in dict1.items():
... # If the word is found in the sublist
... if word in sublist:
... # If the current word is found in dict2's keys
... if word in dict2:
... # Append the current key as a one element tuple
... dict2[word] += (key,)
... else:
... # Create a one element tuple and assign it to the word
... dict2[word] = (key,)
...
>>> dict2
{'item': (1, 2, 3), 'word': (1, 2), 'thing': (1, 3)}
如果你知道字典理解,那么同样可以写成
>>> {word: tuple(k for k, v in dict1.items() if word in v) for word in all_words}
{'item': (1, 2, 3), 'word': (1, 2), 'thing': (1, 3)}
整个元组创建逻辑,基于每个相应word
的dict1
,被挤压为单个生成器表达式并转换为带元tuple(k for k, v in dict1.items() if word in v)
的元tuple(k for k, v in dict1.items() if word in v)
你的代码的逻辑是不正确的,因为你只是迭代3个对象,而你只需要遍历你的字典并反转键和值的位置但是因为你可能有重复的值你可以使用set
容器来保存每个对应的键。名称。 dict.setdefault
是这种情况的一个很好的工具:
>>> d={}
>>> for i,j in dict1.items():
... for k in j:
... d.setdefault(k,set()).add(i)
...
>>> d
{'item': set([1, 2, 3]), 'word': set([1, 2]), 'thing': set([1, 3])}
问题是你正在循环dict2.items
而它应该是dict1.items
。 如果找到,您dict2
键附加到dict2
值中,只需将值重新分配给dict1
值中的最后一个键dict1
。 因此dict2
值不是您所期望的。
相反,您也可以使用collections.defaultdict
(或使用@Kasra的解决方案,@ thefourtheye):
from collections import defaultdict
dict2 = defaultdict(tuple)
for word in all_words:
for key, sublist in dict1.iteritems(): # this
if word in sublist:
dict2[word] += (k,)
else:
dict2[word] = (k,)
dict2
Out[3]: defaultdict(<type 'tuple'>, {'item': (1, 2, 3), 'word': (1, 2), 'thing': (1, 3)})
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.