python——numpy中的size()函数

官方文档截图:

numpy.size

介绍

size()函数:用来统计矩阵元素个数,或矩阵某一维上的元素个数的函数。

参数

numpy.size(a, axis=None)
a:输入的矩阵
axis:int型的可选参数,指定返回哪一维的元素个数。当没有指定时,返回整个矩阵的元素个数。

代码示例

#导入模块
In [5]: import numpy as np

#创建数组
In [6]: t = np.arange(1,13).reshape(3,4)

#使用size函数
In [7]: num = np.size(t)

#得到结果
In [8]: num
Out[8]: 12

#使用size函数,传参数0,表示行
In [9]: num = np.size(t,0)

#得到结果,总共有三行
In [10]: num
Out[10]: 3

#使用size函数,传参数1,表示列
In [11]: num = np.size(t,1)

#得到结果
In [12]: num
Out[12]: 4

我们可以看到,size的axis不设定也行,设定 0,1也可以,当然现在我们的例子是二维数组,如果是三维数组的话,还可以传入2

#创建三维数组
In [18]: t = np.arange(1,19).reshape(2,3,3)

#输出打印数组 t 
In [19]: t
Out[19]:
array([[[ 1,  2,  3],
        [ 4,  5,  6],
        [ 7,  8,  9]],

       [[10, 11, 12],
        [13, 14, 15],
        [16, 17, 18]]])

#不传入参数,默认返回矩阵的元素个数 
In [20]: np.size(t)
Out[20]: 18

#当数组是三维数组时,传入 0 表示数组的 块
In [21]: np.size(t,0)
Out[21]: 2

#表示数组的行
In [22]: np.size(t,1)
Out[22]: 3

#表示数组的列
In [23]: np.size(t,2)
Out[23]: 3

总结

二维数组

axis作用
None返回矩阵的元素个数
0返回该二维矩阵的行数
1返回该二维矩阵的列数

三维数组

axis作用
None返回矩阵的元素个数
0返回该三维矩阵的块(几个二维数组)
1返回二维矩阵的行数
2返回二维矩阵的行数

完。


版权声明:本文为qq_43212169原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。