python计算分段函数值的方法

本博文源于python科学计算，旨在解析如何使用python进行计算分段函数值。下面就以复杂的二元函数进行演示。

题目再现

求解思路

首先先生成x1,x2，然后我们将其映射到网格里，最后写出表达式，进行求解。题目效果，再来张图片，就更加好了！

代码再现

# -*- coding:UTF-8 -*-
# 开发时间：2021/9/1 13:27

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(-2,2,0.04)
y = np.arange(-2,2,0.04)
# 根据x和y生成网格点
[X,Y] = np.meshgrid(x,y)

Pxy = 0.5457*np.exp(-0.75*Y**2 - 3.75*X**2 - 1.5*X) * (X+Y>1) + \
     0.7575*np.exp(-Y**2-6*X**2)*((X+Y<=1) & (X+Y>-1)) + \
     0.5457*np.exp(-0.75*Y**2 - 3.75*X**2 + 1.5*X)*(X+Y<=-1)

# 计算分段函数值
print(Pxy)

# 绘制出结果
fig = plt.figure()
ax = plt.gca(projection="3d")

ax.plot_surface(X,Y,Pxy,cmap="rainbow")
ax.set_xlabel("X")
ax.set_ylabel("Y")
ax.set_zlabel("Z")
plt.show()

计算函数值结果

[[4.13780323e-10 7.95789904e-10 1.51222176e-09 ... 8.55819561e-12
 3.43799817e-12 1.35484814e-12]
[4.65976503e-10 8.96174556e-10 1.70298047e-09 ... 1.00270794e-11
 4.02807814e-12 1.58738717e-12]
[5.23499048e-10 1.00680297e-09 1.91320516e-09 ... 1.17105410e-11
 4.70435829e-12 1.85389601e-12]
...
[8.34343054e-13 2.15823201e-12 5.47662686e-12 ... 4.02602659e-09
 2.14422875e-09 1.12837657e-09]
[7.16690905e-13 1.85389601e-12 4.70435829e-12 ... 3.59225426e-09
 1.91320516e-09 1.00680297e-09]
[6.13662224e-13 1.58738717e-12 4.02807814e-12 ... 3.19753414e-09
 1.70298047e-09 8.96174556e-10]]

绘制图形

总结

分段函数值配合定义域是这段代码的亮点，以后写代码时更应该多多揣摩。

来源：https://blog.csdn.net/m0_37149062/article/details/120038188