下面是一个示例代码,展示了如何按字段对结构进行向量化组合:
import numpy as np
# 定义一个结构数组
dt = np.dtype([('name', 'S20'), ('age', int), ('score', float)])
data = np.array([('Alice', 25, 90.5), ('Bob', 30, 85.5), ('Charlie', 28, 95.0)], dtype=dt)
# 获取每个字段的数据
names = data['name']
ages = data['age']
scores = data['score']
# 按字段进行向量化组合
combined = np.column_stack((names, ages, scores))
# 打印结果
print(combined)
运行上述代码,将得到以下输出:
[[b'Alice' b'25' b'90.5']
[b'Bob' b'30' b'85.5']
[b'Charlie' b'28' b'95.0']]
这里使用了NumPy的column_stack
函数,它将多个一维数组按列组合成一个二维数组。通过使用结构数组的字段,我们可以将每个字段的数据提取出来,然后将它们按列组合起来,得到一个二维数组。
上一篇:按字段对多个数组元素进行排序