np.sqrt() 是向量化函数，比 Python 循环快 10–100 倍，支持任意形状数组和多种输入类型，负数输入返回 nan 并触发 RuntimeWarning，不中断执行。

用 np.sqrt() 直接算，别写循环

NumPy 数组的平方根必须用向量化操作，写 Python 循环或列表推导式纯属浪费时间。因为 np.sqrt() 是底层 C 实现的逐元素函数，对任意形状的数组（1D/2D/N-D）自动广播，速度比纯 Python 快 10–100 倍。

• 输入是标量、Python 列表、np.ndarray 都行，输出类型和输入 dtype 对齐（比如 float32 输入 → float32 输出）
• 遇到负数会返回 nan（不是报错），且触发 RuntimeWarning；若需复数结果，得先转成 complex 类型
• 别用 math.sqrt() 或 ** 0.5 作用于整个数组——前者只接受标量，后者在 NumPy 里虽能广播但语义不清、可读性差、且对负数行为不一致

负数输入时 np.sqrt() 返回 nan，不是崩溃

这是最常被误读的一点：很多人看到警告就以为出错了，其实 np.sqrt() 对负实数输入默认静默返回 nan，并打印 RuntimeWarning: invalid value encountered in sqrt。它不会中断执行，但后续计算若依赖该结果（比如求和、均值），可能把 nan 传下去。

• 检查是否含负数：np.any(arr
• 想忽略警告（不推荐）：np.seterr(invalid='ignore')
• 想安全转复数：np.sqrt(arr.astype(complex))，此时 -4 → 2j
• 想截断为 0：np.sqrt(np.clip(arr, 0, None))

np.sqrt() 和 arr ** 0.5 行为基本一致，但有细微差别

两者都支持广播、都返回新数组、都不修改原数组。但在边界 case 上略有不同：

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限

下载

• 对整数 0：结果都是 0.0，没问题
• 对 int64 超大数（如 2**60）：** 0.5 可能因浮点精度先转成 float64 再开方，而 np.sqrt() 在内部会做类型提升优化，结果略更稳定
• 对 inf 或 nan：行为完全一致，都返回对应浮点特殊值
• 性能几乎无差别，但 np.sqrt() 语义明确，代码意图一目了然，建议优先用它

批量处理多维数组时，注意输出 shape 不变

np.sqrt() 是逐元素运算，不改变数组维度或形状。这点容易被忽略，尤其当配合 axis 参数做其他聚合操作（如 np.sum(axis=1)）混用时，误以为它也能降维。

• 输入 shape 是 (3, 4)，输出一定是 (3, 4)；输入是标量 9，输出是标量 3.0
• 如果想对某轴开方后再聚合（比如每行开方再求和），得拆成两步：np.sqrt(arr).sum(axis=1)，不能写成 np.sqrt(arr.sum(axis=1))——那算的是“先求和再开方”
• 广播也照常工作：比如 arr 是 (5, 3)，b 是 (3,)，那么 np.sqrt(arr + b) 合法，np.sqrt(arr) + b 也合法

实际用的时候，最麻烦的不是怎么写，而是忘了检查输入里有没有负数，或者把「对每个元素开方」和「对整行/整列统计值开方」搞混。这两个地方一错，结果就 quietly 错了。

Python免费学习笔记（深入）：立即使用
在学习笔记中，你将探索 Python 的核心概念和高级技巧！