欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-07-13 09:31:57
...
# 画制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()

画出来的图像
使用Matplotlib包画制x^2+y^2三维图像