numpy中添加新的axis维度(np.newaxis)

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)
发布了19 篇原创文章 · 获赞 0 · 访问量 1143

猜你喜欢

转载自blog.csdn.net/Jinyindao243052/article/details/104211827