修改数组的维度

#导入numpy模块
import numpy as np 
#通过reshape将一维数组修改为二、三维
#创建一个一维数组
a = np.arange(1,25)
print(a)
#将一维修改为二维(2,12) (4,6) (3,8)
b = a.reshape((4,6))
print(b)
#将一维数组修改为三维(2,3,4)
c = a.reshape((2,3,4))
print(c)

#通过np.reshape()进行修改
bb = np.reshape(a,(3,8)) #将一维修改为二维
print(bb)
#将一维修改为三维
cc = np.reshape(a,(4,3,2))
print(cc)

#将多维数组修改为一维数组
d = cc.reshape(24)
print(d)
dd = bb.reshape(-1)
print(dd)

#通过ravel、flatten函数将多维数组转换维一维数组
#flatten
e = bb.flatten()
print(e)
#ravel
f = bb.ravel()
print(f)

[ 1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24]
[[ 1  2  3  4  5  6]
 [ 7  8  9 10 11 12]
 [13 14 15 16 17 18]
 [19 20 21 22 23 24]]
[[[ 1  2  3  4]
  [ 5  6  7  8]
  [ 9 10 11 12]]

 [[13 14 15 16]
  [17 18 19 20]
  [21 22 23 24]]]
[[ 1  2  3  4  5  6  7  8]
 [ 9 10 11 12 13 14 15 16]
 [17 18 19 20 21 22 23 24]]
[[[ 1  2]
  [ 3  4]
  [ 5  6]]

 [[ 7  8]
  [ 9 10]
  [11 12]]

 [[13 14]
  [15 16]
  [17 18]]

 [[19 20]
  [21 22]
  [23 24]]]
[ 1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24]
[ 1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24]
[ 1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24]
[ 1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24]