Python Numpy recarray双向排序

Question

我有一个结构化的numpy数组，并按顺序对其进行排序。它工作得很好，但只有一个方向！

降序：

sort(myStructuredArray,order=my_order)[::-1]

和

升序：

sort(myStructuredArray,order=my_order)

order my_order类似于[col1,col2,-col3,col4,-col5,...,colN]，对于某些列，我想将其按升序排列，如col1、col2和colN；对于其他列，我想按降序排列col3和col5 (减号)。在本例中，我想先按col1升序排序，然后按col2升序排序，然后按col3降序排序，然后按col4升序排序，然后按col5降序排序，依此类推……我该怎么做呢？

谢谢

Answer 1

您可以使用numpy.lexsort实现这一点

In [1]: import numpy as np
In [2]: a = np.array([(4,0), (1,9), (1,0), (4,9)],
                     dtype=[('x',int),('y',float)])

In [3]: a
Out[3]: 
array([(4, 0.0), (1, 9.0), (1, 0.0), (4, 9.0)], 
      dtype=[('x', '<i8'), ('y', '<f8')])

In [4]: a['x']
Out[4]: array([4, 1, 1, 4])

In [5]: a['y']
Out[5]: array([ 0.,  9.,  0.,  9.])

lexsort参数的顺序优先级与np.sort(..., order=...)的相反。因此，要先按'x'的降序排序，然后按'y'的升序排序

In [6]: a[np.lexsort((a['y'], -a['x']))]
Out[6]: 
array([(4, 0.0), (4, 9.0), (1, 0.0), (1, 9.0)], 
      dtype=[('x', '<i8'), ('y', '<f8')])

备注：

• 假设你所有的值都是数值的(因为负数不会颠倒字符串的排序)，这是可行的。
• 我曾在某处看到使用a['x'][::-1]代替-a['x']作为密钥，但现在对我不起作用。