将dtype对象的numpy数组转换为复杂的dtype

Question

我有一个numpy数组，我想把它从一个对象转换成复杂的数组。如果我将该数组作为dtype字符串进行转换，则不会有任何问题：

In[22]: bane
Out[22]: array(['1.000027337501943-7.331085223659654E-6j',
       '1.0023086995640738-1.8228368353755985E-4j',
       '-0.017014515914781394-0.2820013864855318j'], 
       dtype='|S41')

In [23]: bane.astype(dtype=complex)
Out[23]: 
array([ 1.00002734 -7.33108522e-06j,  1.00230870 -1.82283684e-04j,
       -0.01701452 -2.82001386e-01j])

但是当它是dtype对象时，我试图转换它，我得到了一个错误，需要一个浮点数。为什么会这样呢？

In [24]: bane.astype(dtype=object)
Out[24]: 
array(['1.000027337501943-7.331085223659654E-6j',
       '1.0023086995640738-1.8228368353755985E-4j',
       '-0.017014515914781394-0.2820013864855318j'], dtype=object)

In [25]: _.astype(dtype=complex)
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-25-f5d89c8cc46c> in <module>()
----> 1 _.astype(dtype=complex)

TypeError: a float is required    

为了转换它，我使用了两次对astype方法的调用，这看起来很笨拙：

bane_obj
Out[27]: 
array(['1.000027337501943-7.331085223659654E-6j',
       '1.0023086995640738-1.8228368353755985E-4j',
       '-0.017014515914781394-0.2820013864855318j'], dtype=object)

In [28]: bane_obj.astype(dtype=str).astype(dtype=complex)
Out[28]: 
array([ 1.00002734 -7.33108522e-06j,  1.00230870 -1.82283684e-04j,
       -0.01701452 -2.82001386e-01j])

Answer 1

我认为您可能想要执行以下操作，这取决于对象类型内部的内容以及是否没有要担心的填充：

bane.view(np.complex64) or
bane.view(np.complex128)

然而，如果这不起作用，我尝试过的一些小的元组例子就不起作用了，下面的方法起作用了：

bane.astype(np.float).view(np.complex64)

考虑使用numpy结构而不是对象作为基本数据类型，您可能会更容易。