【python】LeetCode刷题 566. Reshape the Matrix

class Solution(object):
    def matrixReshape(self, nums, r, c):
        nums_row=len(nums)
        nums_col=len(nums[0])
        if nums_row*nums_col==r*c:
            tmp=[num for row in nums for num in row]
            col_tmp=[]
            for i in range(0,r):
                row_tmp = []
                for j in range(0,c):
                    row_tmp.append(tmp[i*c+j])
                col_tmp.append(row_tmp)
            return col_tmp
        else:
            return nums
nums =[[1, 2],[3, 4]]
so=Solution()
print so.matrixReshape(nums,1,4)

采用的方法很笨，基本思路是先把列表展开成一个列表，
然后再根据题目要求的维数，把一维列表转换成要求的列表数目。