博客

1. 查找数组元素的长度（字节）

使用Numpy中的itemsize属性，可以获取数组中每个元素的大小（字节）。例如，在以下示例中，我们可以创建一个长度为5的numpy数组，其中每个元素为字符串类型，并使用itemsize属性查找每个元素的长度（字节）。

import numpy as np

# 创建一个字符串类型数组
arr = np.array(['apple', 'banana', 'orange', 'pear', 'grape'])

# 查找每个元素的长度（字节）
print(arr.itemsize)  # 输出：40

上述代码输出的结果为40，这是因为字符串类型中，每个字符占用2个字节，而平均长度为5，所以每个字符串元素的大小为40（即2 * 5 * 2）。

1. 查找数组元素消耗的总字节数

使用Numpy中的nbytes属性，可以获取整个数组元素消耗的总字节数。例如，在以下示例中，我们可以创建一个包含整数和float类型元素的numpy数组，并使用nbytes属性计算整个数组的大小。

import numpy as np

# 创建一个包含整数和float类型元素的数组
arr = np.array([1, 2.3, 4.5, 6, 7.8], dtype=np.float)

# 计算整个数组所需的字节数
print(arr.nbytes)  # 输出：40

上述代码输出的结果为40，这是因为该数组中的每个元素占用了8个字节（float类型），而数组共有5个元素，所以整个数组消耗的字节数为40。