Diyagonal bir matrisi köşegenine göre sırala

Question

Köşelerdeki köşelere sahip olması gereken bir matrisim var ama sütunlar karıştı.

Ama döngü için bariz olmadan verimli bir köşegenleri birlik olsun satırları verişi nasıl bilmiyorum. Sıralamak için hangi anahtarın geçeceğinden bile emin değilim.

Herhangi bir öneriniz var mı?

Answer 1

Sen hedef sütunu sırasını belirlemek ve sütun indisleri olarak argmax sonuçları kullanarak matris yeniden düzenlemek için numpy en argmax kullanabilirsiniz:., Çünkü bunlar,

>>> 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. ]]) 

Aslında sütunlara değil satırlara göre sıralar (ancak sonuç aynıdır). Bu 1 içinde bulunduğu sütunun dizine göre sıralayarak çalışır