numpy中有一个自带的constant(常数):numpy.newaxis
在改变矩阵的shape时十分有用,可为矩阵添加新的维度
import numpy as np
a=np.arange(6)
print(a.shape)
b=a[:,np.newaxis]
c=a[np.newaxis,:]
print('b的shape=',b.shape,'\nc的shape=',c.shape)
运行结果
(6,)
b的shape= (6, 1)
c的shape= (1, 6)
numpy中有一个自带的constant(常数):numpy.newaxis
在改变矩阵的shape时十分有用,可为矩阵添加新的维度
import numpy as np
a=np.arange(6)
print(a.shape)
b=a[:,np.newaxis]
c=a[np.newaxis,:]
print('b的shape=',b.shape,'\nc的shape=',c.shape)
运行结果
(6,)
b的shape= (6, 1)
c的shape= (1, 6)