点击查看代码
# reshape用法
# numpy.arange(n).reshape(a,b)依次生成n个自然数，并且以a行b列的数组形式显示
import numpy as np
array = np.arange(16).reshape(2, 8)
print(array)
# output
# [[ 0 1 2 3 4 5 6 7]
# [ 8 9 10 11 12 13 14 15]]
array1 = np.arange(12).reshape(3, 4)
# array=[a][b]
# array.reshape(m, -1) # 改变维度为m行，d列（其中-1表示列数自动计算，d=a*b/m）
array2 = array1.reshape(4, -1)
# array.reshape(-1, m) # 改变维度为d行，m列（其中-1表示列数自动计算，d=a*b/m）
array3 = array1.reshape(-1, 6)
print(array2)
print(array3)
# output
# [[ 0 1 2]
# [ 3 4 5]
# [ 6 7 8]
# [ 9 10 11]]
# [[ 0 1 2 3 4 5]
# [ 6 7 8 9 10 11]]
