python - 将数据列添加到只有一行的 numpy rec 数组中

Question

我需要将一列数据添加到 numpy rec 数组中。我在这里看到了很多答案，但它们似乎不适用于仅包含一行的记录数组...

假设我有一个记录数组x:

>>> x = np.rec.array([1, 2, 3])
>>> print(x)
rec.array((1, 2, 3), 
      dtype=[('f0', '<i8'), ('f1', '<i8'), ('f2', '<i8')])

我想将值 4 附加到具有自己的字段名称和数据类型的新列，例如

 rec.array((1, 2, 3, 4), 
      dtype=[('f0', '<i8'), ('f1', '<i8'), ('f2', '<i8'), ('f3', '<i8')])

如果我尝试使用正常的 append_fields 方法添加列；

>>> np.lib.recfunctions.append_fields(x, 'f3', 4, dtypes='<i8', 
usemask=False, asrecarray=True)

然后我最终得到

TypeError: len() of unsized object

事实证明，对于只有一行的记录数组，len(x) 不起作用，而 x.size 可以。如果我改用np.hstack()，我会得到TypeError: invalid type Promotion，如果我尝试np.c_，我会得到一个不想要的结果

>>> np.c_[x, 4]
array([[(1, 2, 3), (4, 4, 4)]], 
  dtype=(numpy.record, [('f0', '<i8'), ('f1', '<i8'), ('f2', '<i8')]))

Answer 1

创建初始数组，使其形状为 (1,)；注意额外的括号:

In [17]: x = np.rec.array([[1, 2, 3]])

(如果x是您无法以这种方式控制的输入，则可以在中使用它之前使用x = np.atleast_1d(x) append_fields().)

然后确保 append_fields 中给出的值也是长度为 1 的序列:

In [18]: np.lib.recfunctions.append_fields(x, 'f3', [4], dtypes='<i8', 
    ...: usemask=False, asrecarray=True)
Out[18]: 
rec.array([(1, 2, 3, 4)], 
          dtype=[('f0', '<i8'), ('f1', '<i8'), ('f2', '<i8'), ('f3', '<i8')])