numpy size()函数| Python

在Python中， numpy.size()函数计算沿给定轴的元素数。

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. 
 

代码#1：

Python3

# Python program explaining
# numpy.size() method
 
# importing numpy
import numpy as np
 
# Making a random array
arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])
 
# By default, give the total number of elements.
print(np.size(arr))

输出：

8

代码#2：

Python3

# Python program explaining
# numpy.size() method
 
# importing numpy
import numpy as np
 
# Making a random array
arr = 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-axis
print(np.size(arr, 0))
 
#gives no. of columns along y-axis
print(np.size(arr, 1))

输出：

2
4