从Numpy数组中选择索引的快速方法

时间:2019-03-08 04:42:24

标签: python numpy indexing

我有一个大型数组X,从中必须选择有趣的indices,如下所示。 X是60000 x 784。我浪费了一个循环,每次调用都花一秒钟。

%timeit subX = X[:,indices]
890 ms ± 23.6 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)

如何更快地访问阵列?索引是list,范围从500到784。

0 个答案:

没有答案