Morten Hjorth-Jensen
39 lines
857 B
Python
39 lines
857 B
Python
import numpy as np
|
|
from random import random, seed
|
|
|
|
|
|
# Make data.
|
|
x = np.arange(0, 1, 0.5)
|
|
y = np.arange(0, 1, 0.5)
|
|
x, y = np.meshgrid(x,y)
|
|
|
|
def FrankeFunction(x,y):
|
|
term1 = 0.75*np.exp(-(0.25*(9*x-2)**2) - 0.25*((9*y-2)**2))
|
|
term2 = 0.75*np.exp(-((9*x+1)**2)/49.0 - 0.1*(9*y+1))
|
|
term3 = 0.5*np.exp(-(9*x-7)**2/4.0 - 0.25*((9*y-3)**2))
|
|
term4 = -0.2*np.exp(-(9*x-4)**2 - (9*y-7)**2)
|
|
return term1 + term2 + term3 + term4
|
|
|
|
|
|
def create_X(x, y, n ):
|
|
if len(x.shape) > 1:
|
|
x = np.ravel(x)
|
|
y = np.ravel(y)
|
|
|
|
N = len(x)
|
|
l = int((n+1)*(n+2)/2) # Number of elements in beta
|
|
X = np.ones((N,l))
|
|
|
|
for i in range(1,n+1):
|
|
q = int((i)*(i+1)/2)
|
|
for k in range(i+1):
|
|
X[:,q+k] = (x**(i-k))*(y**k)
|
|
|
|
return X
|
|
|
|
# Making meshgrid of datapoints and compute Franke's function
|
|
p = 2
|
|
z = FrankeFunction(x, y)
|
|
X = create_X(x, y, p)
|
|
print(X)
|