python - Numpy 数组、花式索引、复数

Question

以下代码将数组的一部分乘以一个数字

def mul_by_num(a,b):
    a[0:2] *= b


import numpy as np
a = np.ones(5,dtype=np.float64)
mul_by_num(a,1.0)
mul_by_num(a,1j) #Generates a warning (and casts to float!)

第二次调用产生警告

-c:2: ComplexWarning: Casting complex values to real discards the imaginary part

问题是，将 numpy 数组的一部分乘以复数/实数而不弄乱 dtype 的最 Pythonic 方法是什么？我真的不想从一开始就将数组转换为复数，但程序原则上可以获得复数输入。

编辑：

我不关心复制整个数组，将其转换为复杂的；但我想避免检查 dtypes（即 np.float32、np.float64、np.complex、np.int 等）

Answer 1

您将需要在某些时候将数组转换为复数，否则它将无法保存复数。

将数组转换为复数的最简单方法是添加0j：

if (np.iscomplexobj(b)):
    a = a + 0j
a[0:2] *= b

注意：不是 a += 0j因为它会尝试就地修改数组，如果它还不复杂，这将不起作用。

Answer 2

由于提高了计算速度，numpy 数组确保具有相同的类型。也许您可以尝试使用 python 列表或强制转换它。