标签: python numpy
我有一个数组(nXn)。我有另一个数组b (n X 1). 我想从A中提取那些索引是b元素的元素。我想要以下元素[a[1, b[1]], a[2,b[2]] ...]。 有没有办法在没有循环的情况下执行此操作?
(nXn)
b (n X 1).
[a[1, b[1]], a[2,b[2]] ...]
EG: A = [[1,2,3],[4,5,6],[5,2,8]] B = [2,1,2] 我需要以下数组:[A[0, B[0]], A[1, B[1]], A[2, B[2]]] = [3, 5, 8]
A = [[1,2,3],[4,5,6],[5,2,8]]
B = [2,1,2]
[A[0, B[0]], A[1, B[1]], A[2, B[2]]] = [3, 5, 8]