首页 > 试题广场 >

重塑矩阵

[编程题]重塑矩阵

在机器学习中，我们经常需要对矩阵进行重塑（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库。

算法知识视频讲解

这道题你会答吗？花几分钟告诉大家答案吧！

提交观点

线性代数

难度：

0条回答 2收藏 159浏览

重塑矩阵

扫一扫，把题目装进口袋