np.transpose(np.reshape(np.array([np.arange(7)] * 7 * 2,),(2,7,7)),(1,2,0))
前言:在数据预处理时,遇到了以上代码。当时一脸懵逼,一步步刨解下来,才豁然开朗,因此,记录如下。
import numpy as np
a=np.arange(7) #生成一个含有7个元素的一维列表
print(a)
b=np.array([a] * 7 * 2,) #生成14行7列的二维数组
print(b)
#变换为一个[2,7,7]的三维数组:即以7行7列的二维数组作为一个单元,总共有2个。
c=np.reshape(b,(2,7,7))
print(c)
#生成一个[7,7,2]的三维数组:即以7行2列的二维数组作为一个单元,总共有7个。
#注:transpose(c,(1,2,0))理解如下:
#首先对于c而言,c是一个[2,7,7]数组,其对应的索引值为(0,1,2),那么接下来就好理解了:
#transpose(c,(1,2,0))即通过改变索引值,从而改变数组shape [2,7,7]--->[7,7,2]
d=np.transpose(c,(1,2,0))
print(d)