根據其對角線排列一個numpy矩陣

Question

我有一個矩陣，它應該在對角線上有一個矩陣，但列是混合起來的。

但我不知道怎麼回事，沒有明顯的循環，有效地互換行以獲得在對角線上的統一。我甚至不知道我會通過什麼樣的關鍵排序。

有什麼建議嗎？

Answer 1

您可以使用numpy的的argmax確定了球門立柱訂購和使用argmax結果列索引重新排列矩陣：

>>> z = numpy.array([[ 0.1 , 0.1 , 1. ], 
...     [ 1. , 0.1 , 0.09], 
...     [ 0.1 , 1. , 0.2 ]]) 

numpy.argmax(z, axis=1) 

>>> array([2, 0, 1]) #Goal column indices 

z[:,numpy.argmax(z, axis=1)] 

>>> array([[ 1. , 0.1 , 0.1 ], 
...  [ 0.09, 1. , 0.1 ], 
...  [ 0.2 , 0.1 , 1. ]]) 

Answer 2

>>> import numpy as np 
>>> a = np.array([[ 1. , 0.5, 0.5, 0. ], 
...    [ 0.5, 0.5, 1. , 0. ], 
...    [ 0. , 1. , 0. , 0.5], 
...    [ 0. , 0.5, 0.5, 1. ]]) 
>>> np.array(sorted(a, cmp=lambda x, y: list(x).index(1) - list(y).index(1))) 
array([[ 1. , 0.5, 0.5, 0. ], 
     [ 0. , 1. , 0. , 0.5], 
     [ 0.5, 0.5, 1. , 0. ], 
     [ 0. , 0.5, 0.5, 1. ]]) 

它實際上按行排序，而不是列（但結果是相同的）。它的工作原理是由列的1是指數排序