gpt4 book ai didi

python - 将 dtype 对象的 numpy 数组转换为 dtype complex

转载 作者:太空狗 更新时间:2023-10-29 21:11:02 30 4
gpt4 key购买 nike

我有一个 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 对象并且我尝试转换它时,我得到一个错误,需要一个 float。这是为什么?

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])

最佳答案

我认为您可能想要执行以下操作,具体取决于对象类型以及是否无需担心填充:

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

但是,如果这不起作用,我尝试过的一些小元组示例则不起作用,则以下方法有效:

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

考虑使用 numpy 结构而不是对象作为基本数据类型,总的来说你可能会更轻松。

关于python - 将 dtype 对象的 numpy 数组转换为 dtype complex,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29042512/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com