选择二维数组的二维特定元素

快速的方法 zip 是:

a = [[5, -4, 6], [9, 0, 6], [20, -1, 9], [6, 5, -7]]
b = [0, 1, 2, 0]
result = [ai[bi] for ai, bi in zip(a, b)]
print(result)
# [5, 0, 9, 6]

如果你被允许使用 numpy ,您可以:

>>> import numpy as np
>>> a = [[5, -4, 6], [9, 0, 6], [20, -1, 9], [6, 5, -7]]
>>> b = [0, 1, 2, 0]
>>> arr = np.array(a)
>>> arr[np.arange(len(a)), b]
array([5, 0, 9, 6])

a = [[5, -4, 6], [9, 0, 6], [20, -1, 9], [6, 5, -7]]
b = [0, 1, 2, 0]

c = list(a[i][b[i]] for i in range(len(a)))
#or c = list([ai[bi] for ai, bi in zip(a, b)])
print(c) #return [5, 0, 9, 6]

试用使用 map 在里面使用 lambda 和 zip :

a = [[5, -4, 6], [9, 0, 6], [20, -1, 9], [6, 5, -7]]
b = [0, 1, 2, 0]
print(list(map(lambda x: x[0][x[-1]], list(zip(a,b)))))

输出:

[5, 0, 9, 6]