leetcode

class :
    def matrixReshape(self, nums: List[List[int]], r: int, c: int) -> List[List[int]]:
        h = len(nums)
        w = len(nums[0])
        if (r*c != h*w):
            return nums
        else:
            ans = []
            for _ in range(r):
                ans.append([0] * c)
            for i in range(h):
                for j in range(w):
                    
                    ans[(i*w+j)//c][(i*w+j)%c] = nums[i][j]
            return ans