python - 生成反向索引

Question

我有以下内容：

strlist = ['the', 'the', 'boy', 'happy', 'boy', 'happy']
{x:{(list(enumerate(strlist))[y])[0]} for y in range(len(strlist)) for x in (strlist)}

我的输出如下：

{'boy': set([5]), 'the': set([5]), 'happy': set([5])}

我的问题是我想输出这个（使用python 3.x）：

{'boy': {2,4}, 'the': {0,1}, 'happy': {3,5} }

任何帮助都会很棒！

谢谢

Answer 1

>>> strlist = ['the', 'the', 'boy', 'happy', 'boy', 'happy']
>>> from collections import defaultdict
>>> D = defaultdict(set)
>>> for i, s in enumerate(strlist):
...     D[s].add(i)
... 
>>> D
defaultdict(<type 'set'>, {'boy': {2, 4}, 'the': {0, 1}, 'happy': {3, 5}})

如果defaultdict由于某种原因无法使用

>>> D = {}
>>> for i, s in enumerate(strlist):
...     D.setdefault(s, set()).add(i)
... 
>>> D
{'boy': {2, 4}, 'the': {0, 1}, 'happy': {3, 5{}

这是将其编写为理解的愚蠢（低效）方式

>>> {k: {i for i, j in enumerate(strlist) if j == k} for k in set(strlist)}
{'boy': {2, 4}, 'the': {0, 1}, 'happy': {3, 5}}

Answer 2

尝试

dict(((string, set(i for i,w in enumerate(strlist) if w == string)) for string in strlist))

但请注意，它具有二次运行时间，因此它仅对非常少量的数据有用。

测试用例和示例输出：http : //ideone.com/4sxUNf