從Python中的比較矩陣獲取數據

Question

我有一個從給定的列表返回一個比較矩陣的功能：

def compare(a, b): 
    if b > a: 
     return 1 
    elif b < a: 
     return -1 
    else: 
     return 0 

def matrix(data): 
    return [[compare(a, b) for b in data] for a in data] 

我以這種方式使用此功能：

>>> matrix([0, 4, 5, 2, 1, 3]) 
[[0, 1, 1, 1, 1, 1], 
[-1, 0, 1, -1, -1, -1], 
[-1, -1, 0, -1, -1, -1], 
[-1, 1, 1, 0, -1, 1], 
[-1, 1, 1, 1, 0, 1], 
[-1, 1, 1, -1, -1, 0]] 

我需要一個函數返回給定矩陣的數據，如下面的代碼，但我不知道該怎麼做。

>>> data_from_matrix([[0, 1, 1, 1, 1, 1], 
         [-1, 0, 1, -1, -1, -1], 
         [-1, -1, 0, -1, -1, -1], 
         [-1, 1, 1, 0, -1, 1], 
         [-1, 1, 1, 1, 0, 1], 
         [-1, 1, 1, -1, -1, 0]]) 
[0, 4, 5, 2, 1, 3] 

Answer 1

一個簡單的黑客是計算在矩陣中的每一行的總和：

def data_from_matrix(m): 
    return [(len(m) - 1 - sum(row)) // 2 for row in m] 

這假定矩陣實際上定義了一個總的排序和不檢查一致性的矩陣。另一個假設是設置總排序應該定義爲range(len(m))。

例子：

>>> data_from_matrix([[ 0, 1, 1, 1, 1, 1], 
...     [-1, 0, 1, -1, -1, -1], 
...     [-1, -1, 0, -1, -1, -1], 
...     [-1, 1, 1, 0, -1, 1], 
...     [-1, 1, 1, 1, 0, 1], 
...     [-1, 1, 1, -1, -1, 0]]) 
[0, 4, 5, 2, 1, 3]