python使用Matplotlib包画制x^2+y^2三维图像

最新推荐文章于 2024-12-09 11:18:59 发布

原创最新推荐文章于 2024-12-09 11:18:59 发布 · 3.4k 阅读

5 ·

本内容遵循CC 4.0 BY-SA版权协议

By 张小鸿

收录于

深度学习

当前文章被收录于：

深度学习

8 篇文章 4 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文展示了如何使用Python的Matplotlib库和NumPy创建并保存一个三维图像，通过X, Y网格计算并绘制x^2 + y^2的曲面。教程包括了生成网格数据、计算Z值以及使用plot_surface函数展示结果。

低功耗蓝牙项目，需要一块懂省电的板

思澈 SF32LB52 芯片，BLE 协议栈深度优化，上手即开发

点击查看

# 画制x1^2 + x2^2
from matplotlib import pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = Axes3D(fig)
X = np.arange(-4, 4, 0.25)
Y = np.arange(-4, 4, 0.25)
X, Y = np.meshgrid(X, Y)
# R = np.sqrt(X**2 + Y**2)
# Z = np.sin(R)
Z = X**2 + Y**2

# 具体函数方法可用 help(function) 查看，如：help(ax.plot_surface)
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')
plt.savefig('D:\study\AI\deep_learning_book\deep-learning-from-scratch-master\ef3d.png') #保存图像
plt.show()