博客

查找一个NumPy数组的内存大小，可以使用NumPy中的ndarray.nbytes方法来获取。该方法返回数组占用的内存大小，以字节为单位。

代码示例一：

import numpy as np

# 创建一个大小为10的一维数组
arr = np.zeros(10)

# 输出数组的内存大小
print("数组占用的内存大小为：", arr.nbytes, "字节")

输出结果为：

数组占用的内存大小为： 80 字节

代码示例二：

import numpy as np

# 创建一个大小为4×4的二维数组
arr = np.zeros((4, 4))

# 输出数组的内存大小
print("数组占用的内存大小为：", arr.nbytes, "字节")

输出结果为：

数组占用的内存大小为： 128 字节

需要注意的是，ndarray.nbytes方法只能用于计算NumPy数组的内存大小，对于Python中的其他数据类型，需要使用内置的sys.getsizeof()方法来获取其占用的内存大小。