给定二维矩阵nums,将其转化为r行c列的新矩阵。若无法完成转化,返回原矩阵。
直接调用numpy
import numpy as np
class Solution(object):
def matrixReshape(self, nums, r, c):
try:
return np.reshape(nums, (r, c)).tolist()
except:
return nums
给定二维矩阵nums,将其转化为r行c列的新矩阵。若无法完成转化,返回原矩阵。
直接调用numpy
import numpy as np
class Solution(object):
def matrixReshape(self, nums, r, c):
try:
return np.reshape(nums, (r, c)).tolist()
except:
return nums