温馨提示×

NumPy中怎么创建记录数组

小亿
85
2024-05-11 17:11:50
栏目: 编程语言

在NumPy中,可以使用numpy.rec.array()函数来创建记录数组。记录数组是一种特殊的NumPy数组,其中每个元素都可以包含多个字段。以下是一个示例:

import numpy as np

# 定义字段名和数据类型
dt = np.dtype([('name', 'S10'), ('age', int), ('weight', float)])

# 创建记录数组
arr = np.rec.array([('Alice', 25, 60.5), ('Bob', 30, 70.2)], dtype=dt)

# 访问记录数组的字段
print(arr.name)
print(arr.age)
print(arr.weight)

在上面的示例中,我们首先定义了字段名和数据类型,然后使用np.rec.array()函数创建了一个包含两个记录的记录数组。通过访问记录数组的字段名,我们可以获取对应字段的值。

0