numpy.ndarray.itemsize() 方法 | Python

numpy.ndarray.itemsize() 方法返回每个数组元素的大小（以字节为单位）。

语法

numpy.ndarray.itemsize()

参数

无

返回值

每个数组元素的大小（以字节为单位）。

示例

import numpy as np

arr = np.array([1, 2, 3, 4], dtype=np.float64)

print(arr.itemsize())  # 输出 8

解释

上述示例中，我们创建了一个包含四个浮点数的 numpy 数组。由于我们在创建数组的时候使用了 dtype=np.float64 指定了浮点数的精度，而 np.float64 每个元素大小为 8 字节，所以 arr.itemsize() 返回的值为 8。

适用场景

在使用 numpy 数组时，经常需要获取每个数组元素的大小，比如在计算数组的总大小时需要用到该值。因此，numpy.ndarray.itemsize() 方法非常实用。