将列添加到结构化Numpy数组中

Question

我在geo.dat中有以下数据

id  lon  lat inhab  name
 1   9.  45.   100  Ciriè
 2  10.  45.    60  Acquanegra

我把它装在ndarray里

import numpy as np
data = np.genfromtxt('geo.dat', dtype=None, names=True)

到目前为止，到目前为止，我有一个可以按列名寻址的数据结构。

print(data['name'][1]) #>>> Acquanegra

下一步，问题-我有一个函数，输入两个地理坐标向量(当然是data['LON']和data['LAT'] )，并返回地图上投影位置的两个数组x和y (这是可以的)。

我可以使用独立的向量x和y，但是我想用两个新列data['x']和data['y']来增加data。我天真的尝试

data['x'], data['y'] = convert(data['LON'], data['LAT'])

提出了一个ValueError: no field of name x，教我data有一些字典的特点，但字典不是。

我想怎么做就怎么做？提亚

请考虑一下，.hstack()不适用于结构化数组，也就是记录数组，大多数以前的答案只适用于同构数组(沃伦在下面的评论中提到了这个例外)。

我宁愿不喜欢pandas。

Answer 1

您可以使用np.lib.recfunctions

import numpy.lib.recfunctions as rfn

data = rfn.append_fields(data, ['x', 'y'], [x, y])