将矩阵映射到字典(键:列)

时间:2021-03-30 15:30:34

标签: python dictionary matrix mapping

我正在为这个功能苦苦挣扎,希望您能提供帮助。

给定一个非空矩阵,我需要将其映射到列表字典,因为每个项目的键是第 J 列 IE。 矩阵:

[[1,2,3][4,5,6],[7,8,9,10]]

字典:

{0:[1,4,7],1:[2,5,8],2:[3,6,9],3:[10]}

定义:

matrixToDict(mat)

1 个答案:

答案 0 :(得分:0)

def matrixToDict(mat):
   list2 = []
   maxLength = max(len(x) for x in mat)
   for j in range(maxLength):
       list1 = []
       for i in range(len(mat)):
           if (len(mat[i])-1) >= j:
               list1.append(mat[i][j])
       list2.append(list1)   
   dic = { i : list2[i] for i in range(0, len(list2) ) }
   return dic

mat = [[1,2,3],[4,5,6],[7,8,9,10]]
matrixToDict(mat)

输出:
{0: [1, 4, 7], 1: [2, 5, 8], 2: [3, 6, 9], 3: [10]}