温馨提示×

NumPy中怎么自定义数据类型

小亿
84
2024-05-11 18:21:50
栏目: 编程语言

在NumPy中,可以使用dtype参数来自定义数据类型。以下是一个示例:

import numpy as np

# 定义自定义数据类型
my_dtype = np.dtype([('name', np.str_, 16), ('age', np.int32), ('height', np.float64)])

# 创建一个包含自定义数据类型的NumPy数组
data = np.array([('Alice', 25, 5.6), ('Bob', 30, 6.0)], dtype=my_dtype)

print(data)

在这个示例中,我们首先定义了一个包含名称、年龄和身高字段的自定义数据类型my_dtype。然后我们使用这个自定义数据类型来创建一个包含两个条目的NumPy数组data。最后我们打印出这个数组。

0