Python绘制复函数3D图的问题排查与修正请求

核心问题分析 #

当复函数退化为实函数（即虚部( y=0 )）时，3D图中**( y=0 )的截面必须和对应的2D实函数图像完全重合**。你的侧视图不符，大概率是以下原因导致：

• 3D绘图视角未对准( y=0 )平面，造成视觉偏差
• 坐标轴比例被自动拉伸，扭曲了图像形态
• 未明确标记实函数截面，无法直观对比一致性

修正后的代码（( w=z^2+1 )） #

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# 生成覆盖实函数区间的网格
x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)

# 计算复函数实部：u = x² - y² + 1
U = X**2 - Y**2 + 1

# 创建3D绘图
fig = plt.figure(figsize=(10, 8))
ax = fig.add_subplot(111, projection='3d')

# 绘制3D曲面
ax.plot_surface(X, Y, U, cmap='viridis', alpha=0.7)

# 标记y=0时的实函数截面（对应2D的y=x²+1）
y0_idx = np.where(np.isclose(Y[0], 0))[0][0]
ax.plot(X[:, y0_idx], Y[:, y0_idx], U[:, y0_idx], color='red', linewidth=3, label='y=0时的实函数')

# 设置易懂的坐标轴标签
ax.set_xlabel('实部x')
ax.set_ylabel('虚部y')
ax.set_zlabel('复函数实部u')
ax.set_title('复函数w=z²+1的实部可视化')
ax.legend()

# 切换到侧视图（对准y=0平面，消除视觉偏差）
ax.view_init(elev=30, azim=0)

# 强制坐标轴比例一致，避免拉伸变形
ax.set_box_aspect([1, 1, 1])

plt.show()

修正后的代码（( w=z^3+1 )） #

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

x = np.linspace(-2, 2, 100)
y = np.linspace(-2, 2, 100)
X, Y = np.meshgrid(x, y)

# 计算复函数实部：u = x³ - 3xy² + 1
U = X**3 - 3*X*Y**2 + 1

fig = plt.figure(figsize=(10, 8))
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(X, Y, U, cmap='plasma', alpha=0.7)

# 标记y=0时的实函数截面（对应2D的y=x³+1）
y0_idx = np.where(np.isclose(Y[0], 0))[0][0]
ax.plot(X[:, y0_idx], Y[:, y0_idx], U[:, y0_idx], color='red', linewidth=3, label='y=0时的实函数')

ax.set_xlabel('实部x')
ax.set_ylabel('虚部y')
ax.set_zlabel('复函数实部u')
ax.set_title('复函数w=z³+1的实部可视化')
ax.legend()

# 切换到侧视图
ax.view_init(elev=30, azim=0)
ax.set_box_aspect([1, 1, 1])

plt.show()

关键修正点说明 #

1. 明确标记实截面：添加红色曲线展示( y=0 )时的函数值，直观验证3D图和2D实函数的一致性
2. 固定侧视视角：ax.view_init(elev=30, azim=0) 将视角调整为正对( y=0 )平面，此时看到的截面与2D图完全匹配
3. 统一坐标轴比例：ax.set_box_aspect([1,1,1]) 避免matplotlib自动拉伸坐标轴导致的视觉变形
4. 足够采样密度：使用100个采样点，保证截面曲线光滑，符合高中生的直观认知

高中生讲解提示 #

• 先展示2D实函数图像，再切换到3D图的侧视图，让学生看到红色截面与2D图完全重合，理解复函数是实函数的扩展
• 拖动3D视角，展示( y≠0 )时实部( u )的变化，说明复数引入后，函数从“线”扩展为“面”，体现复数的扩展性

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