numpy.size — NumPy v2.5.dev0 Manual (original) (raw)

numpy.size(a, axis=None)[source]#

Return the number of elements along a given axis.

Parameters:

aarray_like

Input data.

axisNone or int or tuple of ints, optional

Axis or axes along which the elements are counted. By default, give the total number of elements.

Changed in version 2.4: Extended to accept multiple axes.

Returns:

element_countint

Number of elements along the specified axis.

Examples

import numpy as np a = np.array([[1,2,3],[4,5,6]]) np.size(a) 6 np.size(a,axis=1) 3 np.size(a,axis=0) 2 np.size(a,axis=(0,1)) 6