python - 将列表中的项目映射到列表列表

Question

我正在考虑一种将列表中的名称映射到列表列表中的分组索引项的有效方法。

假设我有这样的分组:

g = [[0,1],[2]]

我也有这个列表:

names = ["canine", "dog", "feline"]

我想根据索引将映射的名称返回到分组:

result = [["canine","dog"], ["feline"]]

我不确定如何执行此操作，甚至不确定如何有效执行此操作。这是我目前所拥有的，但没有用。

final = []
for j in range(len(names)):
    for item in g:
        for inner in item:
            res = []
            if inner == j:
                res.append(names[inner])
        final.append(res)
print(final)

如有任何提示，我们将不胜感激。

Answer 1

我认为您可能希望遍历 g 而不是 names。没有理由遍历 names，因为您将使用值 g 对其进行索引。在那种情况下，简单的列表理解似乎更适合这种情况:

[[names[i] for i in sublist] for sublist in g]