In Python, numpy.size() function count the number of elements along a given axis.
Syntax: numpy.size(arr, axis=None)
Parameters:
arr: [array_like] Input data.
axis: [int, optional] Axis(x,y,z) along which the elements(rows or columns) are counted. By default, give the total number of elements in a array
Returns: [int] Return the number of elements along a given axis.
Code #1 :
Python3
# Python program explaining# numpy.size() method# importing numpyimport numpy as np# Making a random arrayarr = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])# By default, give the total number of elements.print(np.size(arr)) |
Output:
8
Code #2 :
Python3
# Python program explaining# numpy.size() method# importing numpyimport numpy as np# Making a random arrayarr = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])# count the number of elements along the axis.# Here rows and columns are being treated# as elements#gives no. of rows along x-axisprint(np.size(arr, 0))#gives no. of columns along y-axisprint(np.size(arr, 1)) |
Output:
2 4
