python之使用另一个数组作为索引从数组中选择元素

说我有一个数组

A = array([[1,2,3], 
           [4,5,6], 
           [7,8,9]]) 

B = array([[1], # want [0, 1] element of A 
           [0],  # want [1, 0], element of A 
           [1]])  # want [2, 1] elemtn of A 

C = array([[2], 
           [4], 
           [8]]) 

请您参考如下方法:

为了答案的完整性......花式索引数组被广播到一个共同的形状，所以以下也有效，并且让你免于最后的 reshape :

>>> A[np.arange(3)[:, None], B] 
array([[2], 
       [4], 
       [8]])