NumPy负数数组平方根计算遇阻：cmath及循环调用报错求助

错误原因分析 #

1. cmath.sqrt不支持数组输入：cmath.sqrt是Python标准库函数，仅能处理单个数值（标量），无法直接作用于NumPy数组，因此触发TypeError: only length-1 arrays can be converted to Python scalars错误。
2. 数组索引语法错误：循环中使用x(i)是错误写法，NumPy数组的索引必须用方括号x[i]，圆括号会被解析为调用函数，导致TypeError: 'numpy.ndarray' object is not callable错误。

正确解决方案 #

优先使用NumPy矢量化运算（比循环效率高得多），以下是两种最优实现：

方法一：使用np.emath.sqrt（推荐）

np.emath.sqrt会自动识别负数并返回复数类型的平方根，无需手动转换数组类型：

import numpy as np

x = np.arange(-10, 10, 0.01)
E = 1

p1 = np.emath.sqrt(E - x**2)

方法二：转换为复数数组后调用np.sqrt

先将计算表达式转为复数类型，再用np.sqrt计算：

import numpy as np

x = np.arange(-10, 10, 0.01)
E = 1

# 将表达式结果转换为复数数组
expr = (E - x**2).astype(np.complex128)
p1 = np.sqrt(expr)

方法三：修正循环实现（不推荐，效率低）

若必须使用循环，需修正索引语法并提前创建结果数组：

import numpy as np
import cmath

x = np.arange(-10, 10, 0.01)
E = 1
# 创建与x同形状的复数类型数组存储结果
p1 = np.empty_like(x, dtype=np.complex128)

for i in range(len(x)):
    p1[i] = cmath.sqrt(E - x[i]**2)

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