matplotlib 绘制等高线

代码

	from matplotlib import colors
	import matplotlib.pyplot as plt
	import numpy as np
	plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
	plt.rcParams['axes.unicode_minus'] = False   # 步骤二（解决坐标轴负数的负号显示问题）
	
	n = 256
	x = np.linspace(-3,3,n)
	y = np.linspace(-3,3,n)
	
	def f(x,y):
	    return (1 - x/2 + x**5 + y**3)*np.exp(-x**2-y**2)
	
	X,Y = np.meshgrid(x,y)
	
	
	plt.figure('等高线')
	
	#颜色
	#8代表分成（8+2）部分
	plt.contourf(X,Y,f(X,Y),8,alpha=0.75,cmap=plt.cm.hot)
	
	#线条
	mainframe = plt.contour(X,Y,f(X,Y),8,colors='black',linewidth=.5)
	
	#添加label
	plt.clabel(mainframe,inline=True,fontsize=10)
	
	
	#隐藏刻度
	plt.xticks(())
	plt.yticks(())
	
	
	plt.show()

效果