reshape的使用
import numpy as np
z = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])
reshape(-1,newshape)的理解 如果reshape的值有-1的话,那么Numpy会根据所给的新的shape的信息(newshape),自动计算补足shape缺的值
比如这里的z矩阵,shape=(3,4),一共有12个元素,如果需要构造2列的新矩阵(只知道新矩阵的列信息), newshape=2,也即是z.reshape(-1,2),那么-1的用处就是根据这个2列这个信息,计算得出新的矩阵需要多少行, 12除以2=6(不能整除会报错),那么新的矩阵就是一个6*2的新矩阵;
再举一个例子,还是这个z矩阵,如果需要构造2乘3小矩阵的三维新矩阵(只知道新矩阵的小矩阵的行列信息), newshape=2乘3,也即是z.shape(-1,2,3),那么-1的用处就是根据这个2乘3小矩阵这个信息,计算得出新的三维矩阵需要多少个2乘3的小矩阵, 12除以(2乘3)=2,那么新的三维矩阵就是一个2乘2乘3的三维矩阵。(这里三维矩阵的说法有问题,希望理解就OK)
z_1=z.reshape(-1)
print("z.reshape(-1)的结果:")
print(z_1)
print("结果大小:",z_1.shape)
z.reshape(-1)的结果:
[ 1 2 3 4 5 6 7 8 9 10 11 12]
结果大小: (12,)
z_2=z.reshape(-1,1)
print("z.shape(-1,1)的结果:")
print(z_2)
print("结果大小:",z_2.shape)
z.shape(-1,1)的结果:
[[ 1]
[ 2]
[ 3]
[ 4]
[ 5]
[ 6]
[ 7]
[ 8]
[ 9]
[10]
[11]
[12]]
结果大小: (12, 1)
z_3=z.reshape(-1,2)
print("z.shape(-1,2)的结果:")
print(z_3)
print("结果大小:",z_3.shape)
z.shape(-1,2)的结果:
[[ 1 2]
[ 3 4]
[ 5 6]
[ 7 8]
[ 9 10]
[11 12]]
结果大小: (6, 2)
z_4=z.reshape(-1,2,3)
print("z.shape(-1,2,3)的结果:")
print(z_4)
print("结果大小:",z_4.shape)
z.shape(-1,2,3)的结果:
[[[ 1 2 3]
[ 4 5 6]]
[[ 7 8 9]
[10 11 12]]]
结果大小: (2, 2, 3)
z_5=z.reshape(-1,3,4)
print("z.shape(-1,3,4)的结果:")
print(z_5)
print("结果大小:",z_5.shape)
z.shape(-1,3,4)的结果:
[[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]]
结果大小: (1, 3, 4)
z_6=z.reshape(-1,2,2,3)
print("z.shape(-1,2,2,3)的结果:")
print(z_6)
print("结果大小:",z_6.shape)
z.shape(-1,2,2,3)的结果:
[[[[ 1 2 3]
[ 4 5 6]]
[[ 7 8 9]
[10 11 12]]]]
结果大小: (1, 2, 2, 3)
|