按 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
想要对数组的 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