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）是所述列索引被提取高級索引。

你的更高級的解決方案是什麼？