用键替换python列表元素

Question

我有一个非唯一字符串列表:

list = ["a", "b", "c", "a", "a", "d", "b"]

我想用一个唯一标识每个字符串的整数键替换每个元素:

list = [0, 1, 2, 0, 0, 3, 1]

该数字无关紧要,只要它是唯一标识符即可.

到目前为止,我所能想到的是将列表复制到一个集合,并使用集合的索引来引用列表.我确信有更好的方法.

Answer 1

这将保证唯一性,并且id是连续的,从0以下开始:

id_s = {c: i for i, c in enumerate(set(list))}
li = [id_s[c] for c in list]

另外,您不应该使用'list'变量名称,因为它会影响内置类型list.

Answer 2

这是一个使用defaultdict的单通道解决方案:

from collections import defaultdict
seen = defaultdict()
seen.default_factory = lambda: len(seen)  # you could instead bind to seen.__len__

In [11]: [seen[c] for c in list]
Out[11]: [0, 1, 2, 0, 0, 3, 1]

这是一种技巧但值得一提!

@ user2357112在相关问题/答案中建议的另一种方法是增加itertools.count.这允许您只在构造函数中执行此操作:

from itertools import count
seen = defaultdict(count().__next__)  # .next in python 2

这可能是优选的,因为default_factory方法不会seen在全局范围内查找.