python - Numpy 字典顺序

Question

我想按字典顺序对以下数组 a 进行排序(获取索引位置)，但是，我在理解 numpy 结果时遇到了问题:

>>> a = np.asarray([[1, 1, 1, 2, 1, 2], [2, 1, 2, 3, 1, 0], [1, 2, 3, 3, 2, 2]])
>>> a
array([[1, 1, 1, 2, 1, 2],
       [2, 1, 2, 3, 1, 0],
       [1, 2, 3, 3, 2, 2]])
>>> np.lexsort(a)
array([0, 5, 1, 4, 2, 3])

例如，我不明白为什么 [1, 2, 1] (a[:,0]) 是 sort-index 0 而 [1, 1, 2] (a[:,1]) 是索引 5，甚至认为它应该比 [1, 2, 1].

Answer 1

键的重要性顺序与您的预期相反。为了获得预期的结果，只需将矩阵倒过来

>>> np.lexsort(np.flipud(a))
array([1, 4, 0, 2, 5, 3])