from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np


def plop(X, Y):
      if X>=1 and X>Y and Y>=0 :
         return (Y/X)-Y
      elif Y>=1 and Y>X and X>=0 :
          return (X/Y)-X
      elif X==Y and Y>0:
          return 1-X
      else:
          return 0


def plop_py(func):
         fig = plt.figure()
         ax = fig.gca(projection='3d')
         myfunc = np.vectorize(func)
         x = fig.gca(projection='3d')
         X = np.arange(-1, 10, 0.05)
         Y = np.arange(-1, 10, 0.05)
         X, Y = np.meshgrid(X, Y)
         Z = myfunc(X,Y)
           
         surf = ax.plot_surface(X, Y, Z, rstride=2, cstride=1,
                     cmap=cm.gist_heat,
                     linewidth=0, antialiased=False, shade=False)
          
         plt.show()


plop_py(plop)