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]）是排序索引0而[1, 1, 2]（a[:,1]）是索引5，甚至认为应该比[1, 2, 1]更轻松。

Answer 1

np.lexsort以字典顺序为您提供列的索引，但是它所考虑的顺序是列中的 last 元素优先于前一个等等。这就是您的示例第5列出现在第1列之前的原因。

[2,0,2] < [1,1,2]因为2 = 2和0 < 1。