NumPy数组运算时能否跳过产生复数（虚数）的计算结果？

要实现跳过复数域输入、继续计算剩余元素的平方根，核心是逐元素遍历计算，配合前置校验或者异常捕获拦截非法输入，避免单个错误中断整个流程，以下是两种可直接复用的实现：

方法1：前置值校验（性能最优） #

实数域内平方根的合法输入为非负数，直接在计算前判断输入值大小，小于0的输入直接标记跳过即可，不会触发任何报错：

import math

def real_domain_sqrt(input_numbers):
    output = []
    for num in input_numbers:
        if num < 0:
            output.append("-")
            continue
        output.append(math.sqrt(num))
    return output

# 测试
x = [-1, 0, 1]
print(real_domain_sqrt(x))
# 运行结果：['-', 0.0, 1.0]

方法2：异常捕获+结果校验（兼容性最强） #

如果你使用的平方根实现会自动给负数返回复数结果（比如cmath.sqrt），或者输入列表可能混有非数值类的非法内容，可以用try/except捕获计算错误，同时校验结果是否为纯实数，适配更多场景：

import math

def robust_sqrt(input_numbers):
    output = []
    for num in input_numbers:
        try:
            res = math.sqrt(num)
            # 适配自动返回复数的sqrt实现时，打开下方注释即可过滤复数结果
            # if isinstance(res, complex) and res.imag != 0:
            #     output.append("-")
            #     continue
            output.append(res)
        except (ValueError, TypeError):
            # 捕获负数传参报错、类型不匹配报错，直接标记跳过
            output.append("-")
    return output

# 测试
x = [-1, 0, 1]
print(robust_sqrt(x))
# 运行结果：['-', 0.0, 1.0]

补充说明 #

• 如果你使用numpy做批量计算，默认numpy.sqrt传入负数不会中断程序，会返回nan并抛出运行时警告，可以通过numpy.isreal()逐元素判断结果，过滤掉非实数值即可。
• 不要直接将整个列表传入无容错逻辑的批量计算函数，必须逐元素处理才能实现单个非法值不阻断整体流程。

内容的提问来源于stack exchange，提问作者mnuizhre