np.newaxis主要用来给数组添加新的维度,直接看下面的使用方法就会明白:
>>>a = np.array([1,2,3,4,5])
>>>a
array([1, 2, 3, 4, 5])
>>>a[:,np.newaxis]
array([[1],
[2],
[3],
[4],
[5]])
>>>a[np.newaxis,:]
array([[1, 2, 3, 4, 5]])
我们下次再见,如果还有下次的话!!!
欢迎关注微信公众号:516数据工作室