Numpy返回索引数组

Question

这可能是一个假问题！但我根本找不到答案！

对于以下数组np.arange(-3,3,1.2)，我得到了这个：

array([-3. , -1.8, -0.6,  0.6,  1.8])

为此我想获得如下的索引数组：

array([0, 1, 2, 3, 4])

提前致谢。

Answer 1

你可以这样做：

a = np.array([-3. , -1.8, -0.6,  0.6,  1.8])
indices = np.arange(a.shape[0])