欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

numpy中函数shape的用法

程序员文章站 2024-02-01 17:18:46
shape函数是numpy.core.fromnumeric中的函数,它的功能是读取矩阵的长度,比如shape[0]就是读取矩阵第一维度的长度。它的输入参数可以使一个整数表示维度,也可以是一个矩阵。这么说你可能不太理解,我们还是用各种例子来说明他的用法: 一维矩阵[1]返回值为(1L,) 二维矩阵, ......

shape函数是numpy.core.fromnumeric中的函数,它的功能是读取矩阵的长度,比如shape[0]就是读取矩阵第一维度的长度。它的输入参数可以使一个整数表示维度,也可以是一个矩阵。这么说你可能不太理解,我们还是用各种例子来说明他的用法:

一维矩阵[1]返回值为(1l,)

>>> z.shape
(1,)

二维矩阵,返回两个值

>>> m = np.zeros((2,3))
>>> m.shape
(2, 3)

一个单独的数字,返回值为空

>>> m = np.zeros(0)
>>> m.shape
(0,)

我们还可以将shape作为矩阵的方法来调用,下面先创建了一个单位矩阵y

>>> y = np.zeros((2,3,4))
>>> y
array([[[0., 0., 0., 0.],
[0., 0., 0., 0.],
[0., 0., 0., 0.]],

[[0., 0., 0., 0.],
[0., 0., 0., 0.],
[0., 0., 0., 0.]]])

我们可以快速读取y的形状

>>> y.shape
(2, 3, 4)

假如我们只想读取y的维度,如下所示:  

>>> z.shape[0]
1
>>> y.shape[0]
2
>>> y.shape[-1]
4
>>> y.shape[-2]
3
>>> y.shape[2]
4
>>> y.shape[1]
3