Plot 3D

Bonjour,

J'essaie de faire un plot 3D d'une fonction. J'ai déjà presque tout le code :

a1, d1 = ThinPlateSplineSmootherCoefficients(t(),z1(t()),0)

fig = plt.figure()
ax = fig.gca(projection='3d')

x = arange(-5, 5, 0.25)
y = arange(-5, 5, 0.25)
x, y = meshgrid(x, y)

surf = ax.plot_surface(x, y, EvaluateThinPlateSpline(u,t(),d1,a1), cmap=cm.coolwarm,
                       linewidth=0, antialiased=False)

ax.set_zlim(-1.01, 1.01)
ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

Mon seul problème est que

u

est un vecteur de R2 qui prend comme première composante

x

et comme deuxième composante

y

, mais si je fais

u = array([x,y])

, on me dit que

u

est un tableau 40x40.

Une idée de comment je pourrais résoudre ce problème?

Merci