题目描述:
代码:
class Solution:
def matrixReshape(self, nums, r,c):
if r*c != len(nums)*len(nums[0]):
return nums
else:
row = 0
col = 0
answer = []
for i in range(0,r):
temp = []
for j in range(0,c):
temp.append(nums[row][col])
col = (col+1)%len(nums[0])
if col == 0:
row += 1
answer.append(temp)
return answer
测试代码:
from solution import *
nums = [[1,2],
[3,4]]
print(Solution().matrixReshape(nums, 1, 4))
测试结果: