我绘制了 z(x,y)等高线,并希望沿等高线提取坐标 x , y 以及相应的坐标 z(x,y)值。
我该怎么做?
以下是一个最小的程序段,用以解释我的要求:
plt.figure()
plt.xlabel('r')
plt.ylabel('z')
plt.xlim(-0.06,0.06)
plt.ylim(-0.06,0.06)
f=open("contourkpt3.txt","w+")
cs=plt.contour(rha,za,ener[:,:,3],[0.15],color=band_color[3])
dat3= cs.allsegs[0][0]
plt.plot(dat3[:,:])
print(dat3[:,:],file=f)
在此代码中,
我不仅要沿轮廓提取rha
,za
坐标,还要获取相应的ener[rha,za,3]
值。
有可能吗?