np.newaxis

最新推荐文章于 2024-06-17 17:48:05 发布

Baymax_HIT

最新推荐文章于 2024-06-17 17:48:05 发布

阅读量109

点赞数 1

分类专栏： python3学习笔记文章标签： numpy

本文链接：https://blog.csdn.net/ggfujfdd/article/details/100608317

版权

python3学习笔记专栏收录该内容

7 篇文章 0 订阅

订阅专栏

本文详细介绍了如何使用NumPy的np.newaxis进行数组维度的扩展，包括从一维到二维，甚至更高维度的转换方法。通过具体代码示例，展示了np.newaxis在不同场景下的应用，帮助读者理解和掌握NumPy数组操作的高级技巧。

摘要由CSDN通过智能技术生成

  x1 = np.array([1, 2, 3, 4, 5])
    # the shape of x1 is (5,)
    x1_new = x1[:, np.newaxis]
    # now, the shape of x1_new is (5, 1)
    # array([[1],
    #        [2],
    #        [3],
    #        [4],
    #        [5]])
    x1_new = x1[np.newaxis,:]	
	# now, the shape of x1_new is (1, 5)
   # array([[1, 2, 3, 4, 5]])

再来一个例子

    In [124]: arr = np.arange(5*5).reshape(5,5)
    
    In [125]: arr.shape
    Out[125]: (5, 5)

# promoting 2D array to a 5D array
In [126]: arr_5D = arr[np.newaxis, ..., np.newaxis, np.newaxis]

In [127]: arr_5D.shape
Out[127]: (1, 5, 5, 1, 1)