from scipy.optimize import curve_fit

import numpy as np

import matplotlib.pyplot as plt

def func(X, a, b, c, d, e):

    x,y = X

    return a*x**2 + b*x + c*y**2 + d*y + e

#데이터 a_x, a_y, a_r

p0 = 1., 1., 1., 1., 1.

a,b,c,d,e = curve_fit(func, (a_x,a_y), a_r, p0)[0]

p_r = func((a_x,a_y), a, b, c, d, e)

X = np.arange(min(a_x), max(a_x), 0.25)

Y = np.arange(min(a_y), max(a_y), 0.25)

X, Y = np.meshgrid(X, Y)

Z = func((X,Y), a, b, c, d, e)

fig1 = plt.figure()

ax = plt.subplot(111, projection = '3d')

ax.scatter(a_x,a_y,a_r)

ax.scatter(a_x,a_y,p_r)

plt.show()

fig1 = plt.figure()

ax = plt.axes(projection='3d')

ax.scatter(a_x,a_y,a_r)

ax.plot_surface(X,Y,Z,color='orange')

plt.show()