将字段添加到结构化numpy数组(2)

import numpy as np
from numpy.lib.recfunctions import append_fields

# 创建一个初始的结构化数组
dtype = [('field1', int), ('field2', float)]
data = np.array([(1, 2.0), (3, 4.0)], dtype=dtype)
print("Original array:")
print(data)

# 定义要添加的新字段及其数据
new_field_name = 'field3'
new_field_data = [5, 6]  # 与原始数组中的元素数量相同
new_field_dtype = int  # 新字段的数据类型

# 向结构化数组中添加新字段
updated_data = append_fields(data, new_field_name, new_field_data, usemask=False, asrecarray=True)
print("
Updated array:")
print(updated_data)

Original array:
[(1, 2.) (3, 4.)]
Updated array:
[(1, 2., 5) (3, 4., 6)]