站点图标 多伦多书苑

Python机器学习手册:从数据预处理到深度学习

1.9 矩阵变形

问题描述

在不改变元素值的前提下,改变一个数组的形状(行数和列数)。

解决方案

使用NumPy的reshape:

讨论

reshape可以重构一个数组,维持该数组原来的数据不变,只改变行数和列数。唯一的要求就是原矩阵和新矩阵包含的元素数量必须相同(也就是大小相同)。可以通过size来查看矩阵的大小:

matrix.size
12

reshape能传入一个非常有用的参数值-1,表示可以“根据需要填充元素”,所以reshape(-1, 1)意味着矩阵的行数是1,而列数则根据需要填充:

最后,如果只提供一个整数作为参数,那么reshape会返回一个长度为该整数值的一维数组:

Author:

退出移动版