||
gird参数后 画出contour等高线图
Cliff Gao
本例子是使用matplotlib做等高线图.
一个学习 scipy的网址:
http://www.scipy-lectures.org/index.html
import matplotlib.pyplot as plt
from matplotlib.mlab import griddata
import numpy as np
def my_contour2d_api(x,y,z,xl,yl,fwn):
""" cliff """
#define grid
xmin,xmax=min(x),max(x)
ymin,ymax=min(y),max(y)
xi=np.linspace(xmin,xmax,100)
yi=np.linspace(ymin,ymax,100)
### grid the data
zi=griddata(x,y,z,xi,yi)
ax=plt.gca()
### contour the gridded data
CS=plt.contour(xi,yi,zi,15) # plot 15 contour lines
CS=plt.contourf(xi,yi,zi,15,cmap=plt.cm.rainbow)
plt.colorbar() #draw colorbar
plt.scatter(x,y,marker='o',c='b',s=5)
#plt.show()
ax.set_xlabel(xl)
ax.set_ylabel(yl)
plt.savefig(fwn)
Archiver|手机版|科学网 ( 京ICP备07017567号-12 )
GMT+8, 2024-9-20 15:17
Powered by ScienceNet.cn
Copyright © 2007- 中国科学报社