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開始