1.9 矩阵变形
问题描述
在不改变元素值的前提下,改变一个数组的形状(行数和列数)。
解决方案
使用NumPy的reshape:
讨论
reshape可以重构一个数组,维持该数组原来的数据不变,只改变行数和列数。唯一的要求就是原矩阵和新矩阵包含的元素数量必须相同(也就是大小相同)。可以通过size来查看矩阵的大小:
matrix.size
12
reshape能传入一个非常有用的参数值-1,表示可以“根据需要填充元素”,所以reshape(-1, 1)意味着矩阵的行数是1,而列数则根据需要填充:
最后,如果只提供一个整数作为参数,那么reshape会返回一个长度为该整数值的一维数组:
本书评论