标签： python numpy vectorization
我有一个numpy矩阵
Ans = [3,5,7]
和普通数组
${workspace_loc}/../../..
我想将矩阵压缩成一列，这样对于第i行，我想选择第B [i]列并将其放入列向量中
所以我想要一个优雅/矢量化的实现来获取
CUR
答案 0 :(得分：1)
试试这个：
In [32]: A[np.arange(A.shape[0]), B] Out[32]: array([3, 5, 7])