python——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	返回二维矩阵的行数

完。