创建一个NumPy记录数组,可以使用NumPy的rec.array
函数。记录数组允许您存储具有不同数据类型的数据,并将它们组织在一个结构化数组中。以下是创建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'])
输出:
['Alice' 'Bob' 'Charlie']
这就是如何创建一个NumPy记录数组的方法。
领取专属 10元无门槛券
手把手带您无忧上云