Question

我想将记录数组中几个字段的内容复制到ndarray（均为float64类型）中。我知道当recarray数据在每个字段中只有一个值时，该怎么做：

my_ndarray[:,0]=my_recarray['X']  #(for field 'X')

现在我有一个rearray，每个字段中都有5个浮点的列表，我只想复制每个列表的第一个元素。当我将上述内容与新的recarray（和列表）一起使用时，出现此错误：

ValueError: could not broadcast input array from shape (92,5) into shape (92)

这是完全有道理的（事后看来）。

我以为我可以得到每个元素的第一个元素：

my_ndarray[:,0]=my_recarray['X'][0]  #(for field 'X')

我收到此错误：

ValueError: could not broadcast input array from shape (5) into shape (92)

我有点理解... numpy仅占用第一行（5个元素）并尝试广播到92个元素的列中。

所以......现在我想知道如何从92个元素列中获取每个列表的第一个元素，划伤我的头... 预先感谢您的建议。

Answer 1

我的猜测是recarray有一个dtype，其中一个字段的形状为5：

In [48]: dt = np.dtype([('X',int,5),('Y',float)])
In [49]: arr = np.zeros(3, dtype=dt)
In [50]: arr
Out[50]: 
array([([0, 0, 0, 0, 0], 0.), ([0, 0, 0, 0, 0], 0.),
       ([0, 0, 0, 0, 0], 0.)], dtype=[('X', '<i8', (5,)), ('Y', '<f8')])

按名称访问此字段将产生一个（3,5）形状的数组（类似于您的（92,5）：

In [51]: arr['X']
Out[51]: 
array([[0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0]])

这可以描述为每条记录的5个项目的列表，但是使用字段名称进行索引会生成一个2d数组，该索引可以像任何2d numpy数组一样进行索引。

让我们将这些值设置为有趣的值：

In [52]: arr['X'] = np.arange(15).reshape(3,5)
In [53]: arr
Out[53]: 
array([([ 0,  1,  2,  3,  4], 0.), ([ 5,  6,  7,  8,  9], 0.),
       ([10, 11, 12, 13, 14], 0.)],
      dtype=[('X', '<i8', (5,)), ('Y', '<f8')])

我们可以使用以下方法获取此字段的第一列：

In [54]: arr['X'][:,0]
Out[54]: array([ 0,  5, 10])

如果您的多个字段具有这样的结构，则可能必须按名称访问每个字段。多字段索引功能是有局限性的。

带列表的recarray：如何引用列表中的第一个元素

1 个答案: