from mpl_toolkits.mplot3d import Axes3D  # noqa: F401 unused import

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

a=np.loadtxt("dipool.out")
nx,ny=np.shape(a)
print (nx,ny)
fig = plt.figure()
p = fig.gca(projection='3d')

X = np.arange(0,nx-1)
Y = np.arange(0,ny-1)
X, Y = np.meshgrid(X, Y)
Z = a[X,Y]
zmax=np.amax(np.amax(a,axis=0))/5
zmin=np.amin(np.amin(a,axis=0))/5


print(zmax,zmin)
lev=np.array([-10,-5,-3,-1,-0.1,0,0.1,1,3,5,10])

surf = p.plot_wireframe(X, Y, Z, color='black',cmap=cm.coolwarm, rcount=20,ccount=20,alpha=0.9, linewidth=1)
cs=p.contour(X,Y,Z,zdir='z',offset=-zmax,levels=lev)
p.set_zlim(zmin, zmax)
fig.colorbar(surf, shrink=0.5, aspect=5)
plt.show()