怎么在python中绘制一个二元函数图像

本篇文章给大家分享的是有关怎么在python中绘制一个二元函数图像，小编觉得挺实用的，因此分享给大家学习，希望大家阅读完这篇文章后可以有所收获，话不多说，跟着小编一起来看看吧。

#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中绘制一个二元函数图像，小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注行业资讯频道。