Numpy索引：获取每列的特定行

Question

例如：

A = np.random.randn(4,32)
inds = np.random.randint(4,size=32)
res = A[inds,np.arange(32)]

这会得到理想的结果，但相当麻烦。是否有更简单的方法来获得这种行为？ （例如使用高级索引）

Answer 1

inds = np.random.randint(4,size=32)
result=[]
for i in range(0,len(inds):
    result.append(A[inds[i],i])
    print result

Answer 2

res = A[inds,np.arange(32)] 

以上是使用numpy高级索引，其中inds是行索引，np.arange（32）是要提取的列索引。

您的更高级解决方案是什么？