1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
import openturns as ot
from matplotlib import pyplot as plt
import openturns.viewer as otv
def C(tau):
return 1.0 / (1.0 + tau * tau)
t0 = 0.0
t1 = 20.0
N = 40
dt = (t1 - t0) / (N - 1)
myMesh = ot.RegularGrid(t0, dt, N)
myCovarianceCollection = ot.SquareMatrixCollection()
for k in range(N):
t = myMesh.getValue(k)
matrix = ot.SquareMatrix(1)
matrix[0, 0] = C(t)
myCovarianceCollection.add(matrix)
covarianceModel = ot.UserDefinedStationaryCovarianceModel(
myMesh, myCovarianceCollection
)
def f(tau):
return [covarianceModel(tau)[0, 0]]
func = ot.PythonFunction(1, 1, f)
func.setDescription(["$t$", "$cov$"])
cov_graph = func.draw(0.0, 20.0, 512)
cov_graph.setTitle("User defined stationary covariance model")
fig = plt.figure(figsize=(10, 4))
cov_axis = fig.add_subplot(111)
otv.View(cov_graph, figure=fig, axes=[cov_axis], add_legend=False)
|
