python - 根据指示第二个列表中值的索引位置从嵌套列表中获取单个元素

Question

作为一个更大项目的最后一部分，这是我要解决的问题:

我有一个列表列表，我需要根据第二个列表的值提取其中的一个元素。

a = [[6,2,3,9], [10,19,14,11], [27,28,21,24]]

b = [0,2,2]

b 中的值表示元素在子列表中的位置。此外，b 中的索引对于列表 a 中的元素索引也是如此。

我要找的结果是:

c = [6, 14, 21]

我试过很多版本:

c = [i[j] for i in a for j in b]

但结果我得到了一个包含所有列表的所有元素的列表，如下所示:

c = [6, 3, 3, 10, 14, 14, 27, 21, 21]

Answer 1

通过使用嵌套循环，您将a 中的每个 元素与b 中的每个 元素组合在一起。您想要的是成对 迭代，使用zip :

c = [x[y] for x, y in zip(a, b)]
# [6, 14, 21]

这大致相当于:

c = [a[i][b[i]] for i in range(min(len(a), len(b)))]