python 中矩阵转换代码的实现

Question

m行n列矩阵amn
其中每一行的数都大于0 并且每行的值相加等于1。
现在想利用python 把这个a_mn 矩阵转化为 r_mn 矩阵。
r_mn 矩阵是 a_mn 在第n行的排序(包括正序，倒序)
例如a_21 = [[0.6,0.3,0.1]， [0.35,0.35,0.3]]
r_21 = [[3,2,1]， [2,2,1]] (正序)
r_21 = [[1,2,3]， [1,1,1]] (倒序)
这样的矩阵转换代码 在python 中如何实现 （任意m，n都需要通用)。

Answer 1

def change(data, sortType):
    test = []
    for item in data:
        result = []
        for value in item:
            l = list(set(item))
            l.sort(reverse=sortType)
            for index1, value1 in enumerate(l, 1):
                if value1 == value:
                    result.append(index1)
        test.append(result)
    return test


data = [[0.6, 0.3, 0.1], [0.35, 0.35, 0.3]]
print(change(data, True))
print(change(data, False))