首页 > 技术文章 > numpy中一维数组增加维度,None,np.newaxis

bravesunforever 2020-04-09 18:57 原文

 

 

#一维数组转化成二维的方法:
np.random.seed(101)
arr=np.random.randint(1,4,size=6)

print("\n原数组:\n",arr)

print("\n如何把一维数组转换成二维的:")
print("arr[:,None]:\n",arr[:,None])
print("")
print("arr[:,np.newaxis]:\n",arr[:,np.newaxis])

 

 

#NO-64,None和np.newaxis都可以
a=np.array([[3,3,3],[4,4,4],[5,5,5]])
b=np.array([1,2,3])

c=b[:,None]  #构造成(3,1)的列向量
d=b[:,np.newaxis]

print(a-c)
print(a-d)

结果:

 

 

 

 

以下是官方文档对None的说明:

官方文档链接

另一个网址:

https://numpy.org/devdocs/reference/arrays.indexing.html

numpy.newaxis
A convenient alias(别名) for None, useful for indexing arrays.

 

 

推荐阅读