将常规 NumPy 数组转换为记录数组,可以使用 NumPy 的 rec.array
函数。记录数组允许您将多个数组合并成一个结构化数组,每个数组元素可以包含多个字段。以下是一个示例:
import numpy as np
# 创建两个常规 NumPy 数组
names = np.array(['Alice', 'Bob', 'Charlie'])
ages = np.array([25, 30, 35])
# 定义记录数组的数据类型
dtype = [('name', 'U10'), ('age', int)]
# 将两个常规数组转换为记录数组
rec_array = np.rec.array(np.zeros(len(names), dtype=dtype))
rec_array['name'] = names
rec_array['age'] = ages
print(rec_array)
输出:
rec.array([('Alice', 25), ('Bob', 30), ('Charlie', 35)],
dtype=[('name', '<U10'), ('age', '<i8')])
在这个例子中,我们首先创建了两个常规 NumPy 数组,分别存储名字和年龄。然后,我们定义了一个记录数组的数据类型,其中包含两个字段:name
和 age
。接下来,我们使用 np.rec.array
函数将两个常规数组转换为记录数组。最后,我们将名字和年龄分别存储在记录数组的 name
和 age
字段中。
这种方法适用于将多个数组转换为记录数组,并允许您轻松地访问和操作结构化数据。
领取专属 10元无门槛券
手把手带您无忧上云