如何用不同列表中的值替换索引？

Question

我有两个清单：

test_list1 = [2, 3, 4, 5, 2, 4] 
test_list2 = [[1, 5], [4, 2, 3], [0]]

我想用test_list2中的值替换test_list1中的索引，这样结果将是：

[[3, 4], [2, 4, 5], [2]]

我试过这个：

res = [test_list1[idx] for idx in test_list2]

我注意到我的代码仅适用于没有子列表的列表

Answer 1

您可以使用递归 function。 这将 go 通过任意嵌套列表替换值。

def replace(lst, replacements):
    for i, val in enumerate(lst):
        if isinstance(val, list):
            replace(val, replacements)
        else:
            lst[i] = replacements[val]

            
test_list1 = [2, 3, 4, 5, 2, 4]
test_list2 = [[1, 5], [4, 2, 3], [0]]

replace(test_list2, test_list1)
print(test_list2)

给出：

[[3, 4], [2, 4, 5], [2]]

Answer 2

具有列表理解和 map function 的 oneliner 可能如下所示：

test_list1 = [2, 3, 4, 5, 2, 4]
test_list2 = [[1, 5], [4, 2, 3], [0]]

output = list(map(lambda x: [test_list1[y] for y in x], test_list2))

print(output)
# [[3, 4], [2, 4, 5], [2]]