python - 如何从 numpy 结构化数组中修改选定行的一列

Question

我正在寻找一种简单的方法来修改它的选定行的 numpy 结构化数组的一个字段。这是我的 SWE：

import numpy as np
dt=np.dtype([('name',np.unicode,80),('x',np.float),('y',np.float)])
a=np.array( [('a',0.,0.),('b',0.,0.),('c',0.,0.) ],dtype=dt)
b=a.copy()
a[a['name']=='a']['x']=1
print a==b # return [ True  True  True]

在这个例子中，a==b结果应该返回[False True True]。实际上，我想从'name'字段中选择我的数组行并修改它的一个字段的值（这里是'x'）。

Answer 1

我找到了答案……重点是场地和面具的位置。您需要将掩码应用于字段列，而不是查找掩码数组的字段：

a['x'][a['name']=='a']=1
print a==b # returns [False  True  True]