Numpy切片后保留一维数组的维度信息

最新推荐文章于 2023-03-07 18:38:40 发布

weixin_30551947

最新推荐文章于 2023-03-07 18:38:40 发布

阅读量1.8k

点赞数 6

文章标签： python

原文链接：http://www.cnblogs.com/redo19990701/p/11469868.html

版权

切片后保留一维数组的维度信息

>>> x
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> a=x[:,0]
>>> a
array([0, 3, 6])
>>> a.shape
(3,)
#丢失维度信息
>>> a=x[:,0,None]
>>> a
array([[0],
       [3],
       [6]])
>>> a.shape
(3, 1)
#未丢失维度信息
>>> a=x[None,:,0]
>>> a
array([[0, 3, 6]])
>>> a.shape
(1, 3)
#未丢失维度信息

转载于:https://www.cnblogs.com/redo19990701/p/11469868.html