我的切片numpy数组的代码(通过花哨的索引)非常慢.它目前是计划的瓶颈.
a.shape
(3218, 6)
ts = time.time(); a[rows][:, cols]; te = time.time(); print('%.8f' % (te-ts));
0.00200009
Run Code Online (Sandbox Code Playgroud)
获取由矩阵a的行'rows'和列'col'的子集组成的数组的正确numpy调用是什么?(事实上,我需要这个结果的转置)