np.newaxis

最新推荐文章于 2024-03-14 09:13:15 发布

qq_23936173

最新推荐文章于 2024-03-14 09:13:15 发布

阅读量743

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： numpy python np

本文链接：https://blog.csdn.net/qq_23936173/article/details/86623455

python 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了numpy库中np.newaxis的使用，该特性用于在一维或二维数组中插入新的维度，以方便进行多维数据操作。通过实例展示了在一维和二维数组上的应用及其输出效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

np.newaxis

np.newaxis的作用是：在当前位置，插入一个新维度。

一维数组

import numpy as np

a = np.arange(5) # a.shape=(5,)
b = a[np.newaxis, :] # b.shape=(1,5)
c = a[:,np.newaxis] # c.shape=(5,1)
d = a[:,np.newaxis,np.newaxis] # d.shape=(5,1,1)
e = a[np.newaxis, np.newaxis, :] # e.shape=(1,1,5)
print('a = ',a)
print('b = ',b)
print('c = ',c)
print('d = ',d)
print('e = ',e)

输出：

a =  [0 1 2 3 4]
b =  [[0 1 2 3 4]]
c =  [[0][1][2][3][4]]
d =  [[[0]][[1]][[2]][[3]][[4]]]
e =  [[[0 1 2 3 4]]]

二维数组

import numpy as np

a = np.arange(6).reshape(2, 3)  # a.shape=(2,3)
b = a[:, np.newaxis]  # b.shape=(2,1,3)
c = a[:, np.newaxis, :]  # c.shape=(2,1,3)
d = a[..., np.newaxis]  # d.shape=(2,3,1)
e = a[np.newaxis, ...]  # e.shape=(1,2,3)
print('a = ', a)
print('b = ', b)
print('c = ', c)
print('d = ', d)
print('e = ', e)

输出：

a =  [[0 1 2]
 	  [3 4 5]]
b =  [[[0 1 2]]
      [[3 4 5]]]
c =  [[[0 1 2]]
      [[3 4 5]]]
d =  [[[0]
       [1]
       [2]]
      [[3]
       [4]
       [5]]]
e =  [[[0 1 2]
       [3 4 5]]]