逐行访问numpy结构化数组

时间:2016-04-18 12:51:23

标签: python arrays numpy structured-array

我试图通过迭代一个字段的值来逐行访问结构化数组,但即使值迭代得很好,数组的切片也不会改变。这是我的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)
for n in a['name']:
  print n,a['name'==n]

给了我:

a (u'a', 0.0, 0.0)
b (u'a', 0.0, 0.0)
c (u'a', 0.0, 0.0)

在每次迭代中,我总是拥有相同的数组切片......奇怪吗?

1 个答案:

答案 0 :(得分:4)

最后一行不对。数组索引的计算结果为True或False,而不是查找命名列。 试试这个:

for n in a['name']:
    print n,a[a['name']==n]