numpy lexsort 用法及代码示例
Lincoln 用法：
numpy.lexsort(keys, axis=-1)
使用键序列执行间接稳定排序。
给定多个排序键(可以将其解释为电子表格中的列)，lexsort返回一个整数索引数组，该数组描述按多个列排序的顺序。序列中的最后一个键用于主排序顺序，second-to-last键用于辅助排序顺序，依此类推。 keys参数必须是可以转换为相同形状的数组的对象序列。如果为keys参数提供了2D数组，则将其行解释为排序键，并根据最后一行，倒数第二行等进行排序。
| – | 描述 |
|---|---|
| 参数 | keys： ： (k, N) array 或 tuple containing k (N,)-shaped sequences
要排序的k个不同的“columns”。最后一列(如果键是2D数组，则为行)是主排序键。 要间接排序的轴。默认情况下，对最后一个轴进行排序。 |
| 返回值： | indices： ： (N，)个整数的ndarray
沿指定轴对键进行排序的索引数组。 |
例子：
姓名排序：首先按姓氏，然后按名称。
>>> surnames = ('Hertz', 'Galilei', 'Hertz')
>>> first_names = ('Heinrich', 'Galileo', 'Gustav')
>>> ind = np.lexsort((first_names, surnames))
>>> ind
array([1, 2, 0])
>>> [surnames[i] + ", " + first_names[i] for i in ind]
['Galilei, Galileo', 'Hertz, Gustav', 'Hertz, Heinrich']
对两列数字进行排序：
>>> a = [1,5,1,4,3,4,4] # First column
>>> b = [9,4,0,4,0,2,1] # Second column
>>> ind = np.lexsort((b,a)) # Sort by a, then by b
>>> ind
array([2, 0, 4, 6, 5, 3, 1])
>>> [(a[i],b[i]) for i in ind]
[(1, 0), (1, 9), (3, 0), (4, 1), (4, 2), (4, 4), (5, 4)]
请注意，排序首先是根据a。二级分类是根据b。
正常argsort会产生：
>>> [(a[i],b[i]) for i in np.argsort(a)]
[(1, 9), (1, 0), (3, 0), (4, 4), (4, 2), (4, 1), (5, 4)]
结构化数组按词法排序argsort：
>>> x = np.array([(1,9), (5,4), (1,0), (4,4), (3,0), (4,2), (4,1)],
... dtype=np.dtype([('x', int), ('y', int)]))
>>> np.argsort(x) # or np.argsort(x, order=('x', 'y'))
array([2, 0, 4, 6, 5, 3, 1])
声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。
