numpy size()函数 | Python

简介

numpy.size()函数用于返回数组中元素的总数量，可以理解为数组的大小或长度。对于多维数组，size()函数返回所有维度中的元素总数。

语法

numpy.size(arr, axis=None)

参数

• arr：输入的数组
• axis（可选）：指定要沿着哪个轴计算大小。默认情况下（axis=None），返回所有元素的总数。如果指定axis=0，则返回矩阵的行数；指定axis=1，则返回矩阵的列数。

返回值

返回整数值，表示数组的大小或长度。

示例

示例 1：计算一维数组的大小

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
size = np.size(arr)
print("Size:", size)

输出结果为：

Size: 5

示例 2：计算二维数组的大小

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
size = np.size(arr)
print("Size:", size)

输出结果为：

Size: 6

示例 3：指定轴计算数组大小

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
size_row = np.size(arr, axis=0)
print("Size along rows:", size_row)

size_col = np.size(arr, axis=1)
print("Size along columns:", size_col)

输出结果为：

Size along rows: 2
Size along columns: 3

总结

numpy.size()函数是一个非常有用的方法，用于计算数组中的元素总数量。它对于确定数组的维度大小或长度非常重要，并可根据需要指定轴来计算沿该轴的大小。numpy中有很多其他的函数和方法可用于操作和处理数组，值得进一步探索。