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

Question

例如：

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

这得到了预期的结果，但相当麻烦。有没有更简单的方法来获得这种行为？（例如，使用高级索引）

Answer 1

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

上面使用的是numpy高级索引，inds是行索引，inds是行索引np.arange公司（32）作为要提取的列索引。你知道吗

你的高级解决方案是什么样的？你知道吗