如何根据索引列表从二维列表中的每个一维列表中选择一个元素？

Question

我们索引列表的第 i 个元素表示我们想知道的 2D 列表中第 i 个 1D 列表的索引。

如果我们的二维列表和索引列表如下所示：

a = [[1,7,3],[9,0,8]]

list_ix = [1, 0]

期望的结果是 [7,9]。

目前我通过使用如下循环来做到这一点：

res = []

for i in range(len(list_ix)):

    res.append(a[i][list_ix[i]])

但这在实际代码中看起来很乱。

有一个更好的方法吗？我将不胜感激任何建议。谢谢你！

Answer 1

我认为这可以通过这种方式完成：res = [a[idx][element] for idx, element in enumerate(list_ix)]

Answer 2

您可以使用zip()功能：a = [[1,7,3],[9,0,8]]    list_ix = [1, 0]out = [l[i] for i, l in zip(list_ix, a)]print(out)印刷：[7, 9]

Answer 3

使用enumeratewhich 可以让您从列表中获取值并跟踪您的当前位置。a = [[1,7,3],[9,0,8]]list_ix = [1, 0]res = []for i, outer_list_index in enumerate(list_ix):    res.append(a[i][outer_list_index])print(res)