计算NumPy数组内存大小的Python方法

在 Python 中，可以使用 numpy 库的 itemsize 属性和 size 属性来计算数组的内存大小。itemsize 返回数组中每个元素的字节大小，size 返回数组中元素的总数。然后，可以将两者相乘来计算整个数组的内存大小。

以下是一个示例代码：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])

memory_size = arr.size * arr.itemsize

print('Memory size of the array:', memory_size, 'bytes')

这将输出数组的内存大小（以字节为单位）。

请注意，这个计算只考虑了数组中元素的内存大小，并不包括任何其他开销，比如数组对象本身的开销。如果你想计算包括这些开销的完整内存大小，可以使用 sys.getsizeof() 函数来获取整个数组对象的大小：

import numpy as np
import sys

arr = np.array([[1, 2, 3], [4, 5, 6]])

memory_size = sys.getsizeof(arr)

print('Memory size of the array:', memory_size, 'bytes')

这将输出整个数组对象的内存大小（以字节为单位）。