Python Numpy size()函数

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

语法: numpy.size(arr, axis=None)
参数:
arr: [array_like] 输入数据。
axis: [int, optional] 轴(x,y,z)，元素(行或列)是沿着轴来计算的。默认情况下，给出一个数组中元素的总数。
返回： [int] 返回沿指定轴线的元素数量。

代码#1：

# 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))

输出:

代码#2：

# 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))