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

使用python绘制二元函数图像的实例

程序员文章站 2022-06-06 21:03:54
废话少说,直接上代码: #coding:utf-8 import numpy as np import matplotlib.pyplot as plt f...

废话少说,直接上代码:

#coding:utf-8
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import axes3d


def function_2(x,y):
   #  这里的函数可以任意定义

   return np.sum(x**2) 
  

fig = plt.figure()
ax = axes3d(fig)
x = np.arange(-3,-3,0.1)
y = np.arange(-3,-3,0.1)
x,y = np.meshgrid(x,y)#创建网格,这个是关键
z = function_2(x,y)
plt.xlabel('x')
plt.ylabel('y')

ax.plot_surface(x,y,z,rstride=1,cstride=1,cmap='rainbow')
plt.show()

以上这篇使用python绘制二元函数图像的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。