2

我有两个具有“兼容”但不同 dtype 的 ndarray,如下所示:

In [22]: A = numpy.empty(shape=(5), dtype=[("A", "f4"), ("B", "f4")])

In [23]: B = numpy.empty(shape=(5), dtype=[("B", "f4"), ("A", "f4")])

In [24]: numpy.concatenate((A, B))
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-24-c930307fb7e4> in <module>()
----> 1 numpy.concatenate((A, B))

TypeError: invalid type promotion

没有明确地逐个字段地连接,有没有办法将两者连接起来?


我是怎么到这种境地的?我不确定,可能是我的代码早期的一个错误,但不管我修复了早期的错误,我都很好奇答案。

4

1 回答 1

2

不是 numpy 专家,但这可能会如您所愿。

import numpy.lib.recfunctions


In [35]: numpy.lib.recfunctions.stack_arrays([A,B], usemask = False)
Out[35]: 
array([(0.0, 0.0), (-3.935500814122861e-10, 4.579443381413502e-41),
       (-4.456803814889554e-10, 4.579443381413502e-41),
       (-8.571845455662697e-09, 4.579443381413502e-41),
       (6.726232628759122e-44, 0.0), (0.0, 0.0),
       (4.579443381413502e-41, -3.935500814122861e-10),
       (4.579443381413502e-41, -4.456803814889554e-10),
       (4.579443381413502e-41, -8.571845455662697e-09),
       (0.0, 6.726232628759122e-44)], 
      dtype=[('A', '<f4'), ('B', '<f4')])
于 2014-06-10T22:32:13.993 回答