首页 > 试题广场 >

重塑矩阵

[编程题]重塑矩阵

热度指数：676 时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 256M，其他语言512M
算法知识视频讲解

在机器学习中，我们经常需要对矩阵进行重塑（reshape）操作。给定一个矩阵，将其重塑为一个新的矩阵，但保持其原始数据。
如果重塑前后的元素总数不相同，返回-1
重塑操作应当满足：
1. 新矩阵的元素总数必须与原矩阵相同
2. 原矩阵中的元素在新矩阵中的相对顺序不变

输入描述:

第一行输入要重塑的矩阵。
第二行输入目标矩阵的行数和列数。

输出描述:

输出重塑后的矩阵，返回形式以list形式。

示例1

输入

[[1, 2], [3, 4]]
(1, 4)

输出

[[1, 2, 3, 4]]

示例2

输入

[[1, 2, 3]]
(5, 1)

输出

-1

说明

重塑前后元素总数不相等，返回-1

备注:

1. 对应的输入、输出已给出，您只用实现核心功能函数即可。
2. 支持numpy、scipy、pandas、scikit-learn库。

算法知识视频讲解

在吵架的四郎很中意你

发表于 2025-04-02 11:33:01

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 展开全文

牛客题解官

发表于 2025-02-05 11:59:09

矩阵重塑是将一个矩阵转换为另一个形状的过程，前提是新形状的元素总数与原矩阵相同。数学表达式为：如果原矩阵 A 的形状为 (m, n)，则重塑后的矩阵 B 的形状为 (p, q)，需要满足：重塑操作可以通过以下代码实现： def reshape_matrix(a: List[List[Union 展开全文

ForHeart

发表于 2025-03-02 17:25:10

from re import A from typing import List, Tuple, Union import numpy as np def reshape_matrix( a: List[List[Union[int, float]]], new_shape: Tuple 展开全文

牛客538341495号

发表于 2025-07-01 11:56:32

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[ 展开全文