如下:
技巧:对于已知的数组维度,比如a为3行4列,如果你想展开成12行一列的话,可以使用a.reshape(12,1)
如果对于未知数组:如果想展开为几行一列,或者一行几列,就可以使用reshape -1 的语法
下面这个例子:想展开成几行一列,’ 几 ‘变为-1,’ 一 ‘变为1
想展开成一行几列,’ 一 ‘变为1,’ 几 ‘变为-1
a=np.arange(1,10)
b1=a.reshape(-1,1)#将数组展开成几行1列的形式,行数=9/1,所以是9行1列
b2=a.reshape(3,-1)#将数组展开成3行几列,列数=9/3,所以变成3行3列
b3=a.reshape(-1,1,3,3)#几部分一页,改变为1部分页3行3列的四维数组
print(b1)
print(b2)
print(b3)
print(b3.shape)
其结果如下:
[[1]
[2]
[3]
[4]
[5]
[6]
[7]
[8]
[9]]
[[1 2 3]
[4 5 6]
[7 8 9]]
[[[[1 2 3]
[4 5 6]
[7 8 9]]]]
(1, 1, 3, 3)