按 python 中的索引对 ndarray 进行排序?

Order sorting of an ndarray by indexes in python?

想要对数组的 np.ndarray 个索引进行排序,例如

[[.5,.7, .9], [.6, .0, .8]]

结果看起来像这样

[[1,1],[0,1],[1,0],[0,1],[1,2],[0,3]]

应用这些索引将获得正确的排序顺序,同时可以应用于与数据匹配的其他结构。

我尝试了 np.argsort,但是没有为 ndarray

提供索引

您可以在平面数组上使用 np.argsort,然后使用 np.divmod 获取您之前形状的索引。 编辑:np.unravel_index 是更高维度的 divmod 替代方案,请参阅 https://numpy.org/doc/stable/reference/generated/numpy.unravel_index.html