题解 | 重塑矩阵
重塑矩阵
https://www.nowcoder.com/practice/87ee1f70415249108a55457065b16b85
from typing import List, Tuple, Union
import numpy as np
def reshape_matrix(a: List[List[Union[int, float]]], new_shape: Tuple[int, int]) -> List[List[Union[int, float]]]:
m = len(a)
n = len(a[0])
num = m*n
out = -1
lista = []
k=0
if num == new_shape[0]*new_shape[1]:
out = [[0]*new_shape[1] for _ in range(new_shape[0])]
for i in range(m):
for j in range(n):
lista.append(a[i][j])
for i in range(new_shape[0]):
for j in range(new_shape[1]):
out[i][j] = lista[k]
k+=1
return out
def main():
try:
a = eval(input())
new_shape = eval(input())
result = reshape_matrix(a, new_shape)
print(result)
except Exception as e:
print(f"输入格式错误: {e}")
if __name__ == "__main__":
main()
