上QQ阅读APP看本书,新人免费读10天
设备和账号都新为新人
4.2.4 拼合、划分一个矩阵
本节以元素为1到16的一个二维矩阵,以及上一节的三维矩阵为例进行介绍。之前已经学过如何用reshape()方法改变矩阵的形状,现在将学习更多方法来改变它的形状以得到某个需要的结果。例如,返回转置矩阵和返回所有元素,代码如下:
输出结果:
矩阵的拼合分为行拼合(即将行拼合在一起使得列数变多,使用np.hstack()方法)和列拼合(即将列拼合在一起使得行数变多,使用np.vstack()方法),示例如下:
输出结果:
矩阵有拼合方法np.vstack()和np.hstack(),也有划分方法np.vsplit()和np.hsplit()。先来看使用行划分方法np.vsplit()的示例,该矩阵必须被均等划分,否则会报“array split does not result in an equal division”错误,代码如下:
输出结果:
或者按照序号划分,示例如下:
输出结果:
再来看使用列划分方法np.hsplit()的示例,代码如下:
输出结果: