import matplotlib.pyplot as plt
from matplotlib import cm
from mayavi import mlab
import numpy as np

a=np.loadtxt("s2d3_last.dat")
#a=np.loadtxt("electr.dat")
zmax=np.amax(np.amax(a,axis=0))
zmin=np.amin(np.amin(a,axis=0))
print(zmax,zmin)
nx,ny=np.shape(a)


X = np.arange(0,nx-1)
Y = np.arange(0,ny-1)
X, Y = np.meshgrid(X, Y)
Z = a[X,Y]*0.1

fig1=mlab.figure(size=(1000,700),bgcolor=(0.5,0.5,0.5))
mlab.mesh(X,Y,Z)
a=mlab.axes(color=(0.9,0.9,0.9),nb_labels=6,zlabel="temperature")
a.axes.font_factor =0.8
a.label_text_property.font_family='times'
s=mlab.colorbar(title='Temperature', orientation='vertical', nb_labels=6)
s.label_text_property.font_family = 'times'
s.label_text_property.bold = 0
s.label_text_property.font_size=12


#fig2=plt.figure()
#plt.contour(X,Y,Z,levels=20)
#plt.show()
mlab.show()