reshape()

点击查看代码

# 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]]