Converting a numpy array of dtype objects to dtype complex
我有一个numpy数组,我想把它从一个对象转换成复杂的。如果我将该数组作为dtype字符串并将其转换,则没有问题:
1 2 3 4 5 6 7 8 9 10 | 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对象并且我试图转换它时,我得到一个错误,即需要一个浮点。这是为什么?
1 2 3 4 5 6 7 8 9 10 11 12 13 | 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方法的调用,这看起来很笨拙:
1 2 3 4 5 6 7 8 9 10 | 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]) |
我想你可能想做以下的事情,这取决于对象类型,如果没有要担心的填充:
1 2 | bane.view(np.complex64) or bane.view(np.complex128) |
但是,如果这不起作用,这对于我尝试的一些小元组示例来说是不起作用的,那么下面的例子就起作用了:
1 | bane.astype(np.float).view(np.complex64) |
号
考虑将numpy结构而不是对象用于基本数据类型,这样做可能会更轻松。