如何裁剪复数python numpy数组中元素的实部和虚部

Question

我有一个数组的复数。我寻求将数组中每个数字的实部和虚部剪辑到某个指定的最小值和最大值（相同的剪辑应用于实部和虚部）。例如，考虑：

import numpy as np

clip_min = -4
clip_max = 3

x = np.array([-1.4 + 5j, -4.7 - 3j])

[-1.4 + 3j, -4-3j]

我可以通过对复数组的实部和虚部调用np.clip，然后将它们相加（在将虚的裁剪数据乘以1j之后）来实现这一点。有办法用一个命令就能做到这一点吗？

np.clip(x, clip_min, clip_max) 

不会产生预期的结果。

Answer 1

有一种比将实部和虚部剪裁为单独的数组更有效的方法，使用就地操作：

np.clip(x.real, clip_min, clip_max, out=x.real)
np.clip(x.imag, clip_min, clip_max, out=x.imag)

如果这些只是存储在复数中的笛卡尔坐标，您可以通过将它们保持为浮点而不是复数来将它们剪辑在单个命令中。

x = array([[-1.4,  5. ],
           [-4.7, -3. ]])
np.clip(x, clip_min, clip_max)
>>> array([[-1.4,  3. ],
           [-4. , -3. ]])