1. 程式人生 > >python——numpy中的size()函式

python——numpy中的size()函式

介紹
size()函式主要是用來統計矩陣元素個數,或矩陣某一維上的元素個數的函式。

引數
numpy.size(a, axis=None)
a:輸入的矩陣
axis:int型的可選引數,指定返回哪一維的元素個數。當沒有指定時,返回整個矩陣的元素個數。

例子

>>> a = np.array([[1,2,3],[4,5,6]])
>>> np.size(a)
6
>>> np.size(a,1)
3
>>> np.size(a,0)
2

axis的值沒有設定,返回矩陣的元素個數
axis = 0,返回該二維矩陣的行數
axis = 1,返回該二維矩陣的列數

注:第二個引數axis從0開始,不是從1開始

參考
官方文件