矩阵乐扣题目

leet刷题

矩阵的重塑&矩阵对角线重排

解题方法的矩阵数据结构不适合numpy模块的矩阵方法
而改成二维数组即可
但是矩阵的对角排列还不行
20200512继续

import numpy as np
from collections import defaultdict
import itertools
mat = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
def weile(mat):
    m,n,d = len(mat),len(mat[0]),defaultdict(list)
    for i,j in itertools.product(range(m),range(n)):
        d[i - j].append(mat[i][j])
    d = {k: iter(sorted(v)) for k,v in d.items()}
    for i,j in itertools.product(range(m),range(n)):
           mat[i][j] = next(d[i -j])
    return mat
def matrixReshape( nums, r, c):
        res=[i for j in nums for i in j]    
        return [res[i:i+c] for i in range(0,len(res),c)]
print(matrixReshape(mat,2,2))
print(weile(mat))