如何根據第一列，然後第2列Theano行進行排序，等

Question

我有一個Theano陣列看起來像這樣：

0 43 1 
4 8 5 
0 41 8 
4 8 4 

我希望基於第一列中的行進行排序，並在第一列中具有相同值的行，根據第二列進行排序等......在這個例子中有3列，但理想情況下，我希望該方法可以處理任意數量的列。

預期的結果是：

0 41 8 
0 43 1 
4 8 4 
4 8 5 

編輯：電池可以有非常大的值（Int64的），所以結合在一起列看起來並不像一個可行的解決方案給我。

Answer 1

假設您的數據的格式一致，您可以簡單地使用sorted進行詞法排序。

假設這是存儲在一個文本文件，你可以做：

with open('foo.txt', 'r') as inputfile: 
    lines = inputfile.read().splitlines() 
    with open('bar.txt', 'w') as outputfile: 
     outputfile.write('\n'.join(sorted(lines))) 

爲什麼？因爲空格出現在詞彙排序的數字之前，所以' 8'小於'10'。 ASCII是專門爲此類排序而設計的。

Answer 2

在numpy中計算Theano的順序列表最好，因爲必須處理綁定的場景，這可能需要if語句。

獲得訂單列表後，將其轉換爲dan​​o ='int64'或'int32'的theano ivector或vector，並將其用作原始矩陣的索引。

例如

0 43 1 
A = 4 8 5 
    0 41 8 
    4 8 4 

計算順序列表中NUMP，從而導致b = [2,0,3,1]

A = theano.matrix() 
order_ls = theano.tensor.ivector(name='orderlist') 
reordered_matrix = A[order_ls] 

得到重新排序矩陣的值，我們可以創建一個像

get_value = theano.function([A, order_ls], reordered_matrix)