jax.numpy.lexsort

jax.numpy.lexsort(keys, axis=-1)

使用一系列键执行间接稳定排序。

LAX-backend 实现的 numpy.lexsort()。

原始文档字符串如下。

给定多个排序键，lexsort 返回一个整数索引数组，该数组描述了按多个键的排序顺序。序列中的最后一个键用于主排序顺序，平局由倒数第二个键打破，依此类推。

参数:

• keys ((k, m, n, ...) array-like) – 要排序的 k 个键。最后一个 键（例如，如果 keys 是二维数组，则为最后一行）是主排序键。沿第零轴的 keys 的每个元素必须是具有相同形状的类数组对象。

• axis (int, optional) – 要间接排序的轴。默认情况下，对每个序列的最后一个轴进行排序。沿 axis 的独立切片分别排序；参见最后一个示例。

返回:

indices – 沿着指定轴对键进行排序的索引数组。

返回类型:

(m, n, …) ndarray of ints