流体模拟中numpy.sqrt遇负数的警告及numpy处理机制问询

• 当传入np.sqrt()的数组包含负数时，NumPy不会终止程序运行，而是会在对应负数的位置返回NaN（Not a Number），同时触发你看到的RuntimeWarning: invalid value encountered in sqrt警告。
• 这种处理逻辑是为了保证数组运算的连续性：哪怕部分元素存在非法输入，整个数组的计算流程仍能完成，不会因单个负数导致程序崩溃。
• 若要避免警告或自定义处理逻辑，可参考以下两种方式：
  1. 提前用布尔索引替换负数为合理值（比如0或NaN）：

     # 将P2中的负数替换为0，避免sqrt报错
     P2_clean = np.where(P2 < 0, 0, P2)
     W2[i, :] = (np.sqrt(P2_clean) / (d2 * v2))
     

  2. 临时禁用该类警告（仅在确认负数不影响最终结果时使用）：

     import warnings
     warnings.filterwarnings('ignore', category=RuntimeWarning)
     

• 如果你的场景需要得到复数结果，可使用np.emath.sqrt()，它会直接返回复数类型的计算结果，不会触发警告：

  W2[i, :] = (np.emath.sqrt(P2) / (d2 * v2))
  

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