如何以字节为单位获得numpy数值类型的大小？

Question

我想得到数字类型的字节大小，特别是numpy定义的那些。这是通过这个辅助函数

实现的

def sizeof(dtype):
    a = np.zeros(1, dtype=dtype)
    return int(a.nbytes)

但我想知道是否会有一种内置的，不那么尴尬的方式。在这种情况下，sys.getsizeof没有任何帮助 - 它为我测试的所有numpy数值类型返回400个字节。

Answer 1

使用dtype.itemsize或ndarray.itemsize：

>>> import numpy as np

>>> arr = np.array([1, 2, 3], dtype=np.int32)
>>> arr.dtype.itemsize
4

>>> arr.itemsize
4