二元二次图像

# 逻辑回归 平方差画图
import numpy as np 
# mpl_toolkits是matplotlib官方的工具包 mplot3d是用来画三维图像的工具包
from mpl_toolkits.mplot3d import Axes3D
# pyplot 是一个有命令风格的的函数集合，与matlab相似。
from matplotlib import pyplot as plt
import math

# 创建一个图像窗口
fig = plt.figure()
# 在图像窗口添加3d坐标轴
ax = Axes3D(fig)

# 使用np.linspace定义 x:范围(-10,10);个数为100
x = np.linspace(-10,10,100)
# 定义 y:范围(-3,3);个数为50
y = np.linspace(-10,10,100)
# 创建x-y平面网络
x,y = np.meshgrid(x,y)
# 定义函数 r=1/2*(x-y)^2
# r = 1/2*np.square(x-y)
r = x**2+y**2-math.sqrt(2)*x*y

# 将函数显示为3d  rstride 和 cstride 代表 row(行)和column(列)的跨度 get_cmap为色图分类 
ax.plot_surface(x,y,r,rstride = 1, cstride = 1,cmap='rainbow')

# 投影
# ax.contourf(x,y,r,zdir= 'x', offset = (-2),cmap=plt.get_cmap('rainbow')) 

# 显示创建的图像
plt.savefig('minus.tif', dpi=200)
plt.show()

实际效果如图：

minus.tif