# -*- coding: Latin-1 -*-
# Programme de visualisation d'une fonction à deux variables réelles
# Dominique Lefebvre pour TangenteX.com
# 17 février 2016
#

# importation des librairies
from scipy import meshgrid, arange
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

 # définition de la fonction
def f(x,y):
    return x**2 + y**2
      
# déclaration des buffers de travail
X = arange(0.0,1.0,0.01)
Y = arange(0.0,1.0,0.01)

# calcul de la fonction
Z = f(X,Y)

# construction de la grille de calcul
SX,SY = meshgrid(X,Y)
    
# affichage du repère et de ses libellés
fig = plt.figure(figsize=(14,8))
ax = fig.add_subplot(111, projection='3d')
ax.set_xlabel('X', fontsize = 16)
ax.set_ylabel('Y', fontsize = 16)
ax.set_zlabel('Z', fontsize = 16)
ax.view_init(elev=15, azim = 120)

# affichage de la fonction
p = ax.plot_surface(SX,SY,Z,rstride=1,cstride=1,linewidth=0, cmap='hot')

# sauvegarde de l'image
plt.savefig('E:\PhysNumWeb1\images\EDP1.png')