快捷導(dǎo)航

numpy多級(jí)排序lexsort函數(shù)的使用

更新時(shí)間：2023年03月27日 15:03:20 作者：地球被支點(diǎn)撬走啦

本文主要介紹了numpy多級(jí)排序lexsort函數(shù)的使用，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

1、lexsort() 排的是個(gè)啥

lexsort() 函數(shù)的定義如下：

def lexsort(keys, axis=None)

lexsort是一種多級(jí)排序方法。作用為對(duì)給定的 keys 中的最后一個(gè) key 排序，每一個(gè) key 都被認(rèn)為是列向量，其他 keys 用來輔助最后一個(gè) key 進(jìn)行排序。最后返回最后一個(gè) key 進(jìn)行排序的索引。

哈哈哈是不是沒看懂上面的解釋？莫慌莫慌，我給舉個(gè)簡(jiǎn)單的例子保證秒懂，可以回想這樣一個(gè)場(chǎng)景：

對(duì)一堆學(xué)生的成績(jī)進(jìn)行排名，則排名結(jié)果就是按照總成績(jī)從低到高排序，這個(gè)還沒排序的總成績(jī)表就是上述說的最后一個(gè) key ，排序后的索引就是上述的返回值。

比如初始總成績(jī)數(shù)組為：[99, 100, 98, 95, 80] ，這就是最后一個(gè) key，升序排序后的數(shù)組為：[80, 95, 98, 99, 100]，對(duì)應(yīng)的索引為：[4, 3, 2, 0, 1]，這個(gè)索引數(shù)組就是 lexsort 函數(shù)的返回值。

但是如果兩個(gè)人的總成績(jī)一樣怎么進(jìn)一步排序呢？這時(shí)可以按照語文成績(jī)的升序進(jìn)行排序，如果語文成績(jī)還一樣呢？那再看數(shù)學(xué)成績(jī)，如果數(shù)學(xué)還一樣呢？再查英語成績(jī)，如果還。。。（沒了沒了，再相同我就要懷疑這倆人作弊了?。┻@里的語文成績(jī)、數(shù)學(xué)成績(jī)、英語成績(jī) 就是最后一個(gè) key 前面的 keys，記得是從后往前數(shù)的。也就是最后一個(gè) key 相同，就看倒數(shù)第二個(gè) key，倒數(shù)第二個(gè)相同就看倒數(shù)第三個(gè) key。

2、舉個(gè)例子

默認(rèn)你已經(jīng)理解了，那就舉一個(gè)例子來過一遍：

import numpy as np
a = [[3, 3, 8],
     [6, 9, 7],
     [3, 3, 2]]
b = np.array(a)  # 轉(zhuǎn)換為 numpy 數(shù)組
"""
index = np.lexsort((b[:, 2], b[:, 1], b[:, 0])) 的運(yùn)行情況：
step-1、首先按照第一列 b[:, 0] 的數(shù)值 [3, 6, 3] 升序順序?qū)γ啃械奈恢眠M(jìn)行重排, 第一行有兩個(gè)值相同, 轉(zhuǎn)到step-2
            [[3, 3, 8],        [[3, 3, 8],
             [6, 9, 7],   -->   [3, 3, 2],   
             [3, 3, 2]]         [6, 9, 7]]
step-2、第一列數(shù)值相同的情況下按照相應(yīng)位置第二列 b[:, 1] 的值繼續(xù)排序, 第二列對(duì)應(yīng)位置任然相同, 轉(zhuǎn)到step-3
            [[3, 3, 8],        [[3, 3, 8],
             [6, 9, 7],   -->   [3, 3, 2],   
             [3, 3, 2]]         [6, 9, 7]]
step-3、第二列數(shù)值相同的情況下按照相應(yīng)位置第三列 b[:, 2] 的值繼續(xù)排序
            [[3, 3, 8],        [[3, 3, 2],
             [6, 9, 7],   -->   [3, 3, 8],   
             [3, 3, 2]]         [6, 9, 7]]
總結(jié): 第一列的排序結(jié)果為[3, 3, 6] 對(duì)應(yīng)的索引值為 [2, 0, 1]
"""
index = np.lexsort((b[:, 2], b[:, 1], b[:, 0]))
print(index)  # 輸出結(jié)果: [2 0 1]
ans = b[index] # 按照排序結(jié)果重置數(shù)組元素的位置
print(ans)
# ans 的結(jié)果：
[[3 3 2]
 [3 3 8]
 [6 9 7]]

如果直接對(duì)矩陣進(jìn)行排序呢？

"""
相當(dāng)于拆解為
key1 = b[0,:]
key2 = b[1,:]
key3 = b[2,:]  # 最后一行為主排序序列
index = np.lexsort(b) 等效為: index = np.lexsort((b[0,:], b[1,:], b[2,:]))
"""
index = np.lexsort(b) 
print(index)  # 輸出結(jié)果: [2 0 1]

即，當(dāng)輸入是一個(gè)矩陣時(shí)，默認(rèn)排序最后一行的數(shù)據(jù)，前面的行都是輔助行。

到此這篇關(guān)于numpy多級(jí)排序lexsort函數(shù)的使用的文章就介紹到這了,更多相關(guān)numpy lexsort函數(shù)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: