如何创建一个numpy记录数组？

创建一个NumPy记录数组，可以使用NumPy的rec.array函数。记录数组允许您存储具有不同数据类型的数据，并将它们组织在一个结构化数组中。以下是创建NumPy记录数组的步骤：

导入NumPy库：

import numpy as np

定义数据类型和数据：

data_dtype = [
    ('id', int),
    ('name', 'U10'),
    ('age', int),
    ('height', float),
    ('weight', float)
]

data = [
    (1, 'Alice', 30, 160.5, 60.0),
    (2, 'Bob', 25, 175.0, 70.0),
    (3, 'Charlie', 22, 170.0, 65.0)
]

在这个例子中，我们定义了一个名为data_dtype的数据类型列表，其中包含每个字段的名称和数据类型。我们还定义了一个名为data的列表，其中包含我们要存储的实际数据。

创建记录数组：

rec_array = np.rec.array(data, dtype=data_dtype)

现在，我们可以使用np.rec.array函数创建一个记录数组，并将我们之前定义的数据和数据类型传递给它。

访问记录数组中的数据：

print(rec_array['name'])

要访问记录数组中的数据，可以使用字段名称作为索引。在这个例子中，我们打印出了所有人的名字。

完整的代码示例：

import numpy as np

data_dtype = [
    ('id', int),
    ('name', 'U10'),
    ('age', int),
    ('height', float),
    ('weight', float)
]

data = [
    (1, 'Alice', 30, 160.5, 60.0),
    (2, 'Bob', 25, 175.0, 70.0),
    (3, 'Charlie', 22, 170.0, 65.0)
]

rec_array = np.rec.array(data, dtype=data_dtype)

print(rec_array['name'])

输出：