Python计算分段函数值与3D可视化完整指南

Ada

发布时间：2026-10-15| 3人浏览过

原创

本文详细演示如何使用Python进行复杂二元分段函数的数值计算与可视化。通过NumPy生成网格点，利用布尔索引处理不同定义域下的表达式，并借助Matplotlib绘制3D曲面图。文章涵盖从环境配置、核心代码实现到结果输出的完整流程，帮助开发者掌握分段函数在科学计算中的应用。

Python分段函数计算与可视化概述

本文基于Python科学计算库，深入解析如何对分段函数进行数值求解。为了更直观地展示处理逻辑，我们将以一个结构较为复杂的二元函数作为具体案例进行演示。

问题描述与解决策略

问题描述

解决策略

核心思路是先利用NumPy生成x1和x2坐标轴数据，随后通过网格化操作生成对应的网格点矩阵。接着，根据分段条件构建数学表达式并完成计算。为了更清晰地展示最终效果，我们将结合数值结果与可视化图形进行说明。

Python代码实现与结果分析

代码实现


# -*- 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]]

图形绘制过程