如何将单词分配给列表中的值？

Question

我有一个包含 4 个不同数字的列表，总共 10 个数字。对于这些数字中的每一个，我想分配一个颜色/单词。 因此，问题是，我如何从一个数字列表转到一个颜色列表，其中每个相同的数字都由相同的颜色/单词表示？

初始列表：

my_list = [72, 50, 3, 50, 16, 72, 3, 72, 3, 50]

预期输出：

print(my_list) -> ['red', 'blue', 'green', 'blue', 'black', 'red', 'green', 'red', 'green', 'blue']

我尝试了什么？

我从 my_list 中提取了唯一编号：

my_list = [72,50,3,50,72,3,72,3,50]

UniqueList = list(set(my_list))
print(UniqueList)

现在我需要从颜色列表 ['red', 'blue', 'green', 'black', 'yellow', 'purple'] 中为这 4 个唯一数字中的每一个分配一个颜色，然后生成my_list 再次使用颜色。不幸的是，我不知道如何前进。有什么想法吗？

Answer 1

您可以创建一个查找字典：

my_list = [72, 50, 3, 50, 16, 72, 3, 72, 3, 50]
colors = ['red', 'blue', 'green', 'black', 'yellow', 'purple']

lookup = dict(zip(set(my_list), colors))
output = [lookup[number] for number in my_list]
print(output)

输出

['red', 'green', 'black', 'green', 'blue', 'red', 'black', 'red', 'black', 'green']

您可以将lookup 字典视为将颜色分配给其中一个数字的函数。在上面的示例中，lookup 具有以下值：

{72: 'red', 16: 'blue', 50: 'green', 3: 'black'}

这意味着每次出现72 时都会分配'red'。你可以像这样自定义lookup：

lookup = {72 : 'green', 50 : 'black', 3 : 'yellow', 16: 'purple'}

这一次每次72 出现green 都会被分配。在here 中查看有关如何创建字典的更多信息。

更新

如果您想保留列表中的出现顺序，您可以执行以下操作：

seen = set()
result = []
for e in my_list:
    if e not in seen:
        result.append(e)
        seen.add(e)

print(result)

输出

[72, 50, 3, 16]

请注意，result 列表按出现顺序排列。

Answer 2

您还可以提供 {int: color} 键值对的字典，然后执行以下操作：

my_list = [72, 50, 3, 50, 16, 72, 3, 72, 3, 50]
mymap = {72: 'red', 50:'blue', 3:'green', 16:'black'}

print([mymap[k]for k in my_list])

## output
['red', 'blue', 'green', 'blue', 'black', 'red', 'green', 'red', 'green', 'blue']