File: t_Staircase_std.py

package info (click to toggle)
openturns 1.26-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 67,708 kB
  • sloc: cpp: 261,605; python: 67,030; ansic: 4,378; javascript: 406; sh: 185; xml: 164; makefile: 101
file content (70 lines) | stat: -rwxr-xr-x 1,779 bytes parent folder | download | duplicates (3)
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#! /usr/bin/env python

import openturns as ot

ot.TESTPREAMBLE()

# Instantiate one distribution object
dim = 1
meanPoint = ot.Point(dim, 1.0)
meanPoint[0] = 0.5
sigma = ot.Point(dim, 1.0)
sigma[0] = 2.0
R = ot.CorrelationMatrix(dim)
distribution1 = ot.Normal(meanPoint, sigma, R)

# Instantiate another distribution object
meanPoint[0] = -1.5
sigma[0] = 4.0
distribution2 = ot.Normal(meanPoint, sigma, R)

# Test for sampling
size = 200
nPoints = 20
sample1 = distribution1.getSample(size)
sample2 = distribution2.getSample(size)

# Construct empirical CDF for each sample
data1 = ot.Sample(nPoints, 2)
data2 = ot.Sample(nPoints, 2)
cursor1 = ot.Point(2)
cursor2 = ot.Point(2)
for i in range(nPoints):
    cursor1[0] = 13.0 * i / nPoints - 6.5
    count1 = 0.0
    cursor2[0] = 24.0 * i / nPoints - 13.5
    count2 = 0.0
    for j in range(size):
        if sample1[j, 0] < cursor1[0]:
            count1 += 1.0
        if sample2[j, 0] < cursor2[0]:
            count2 += 1.0
    cursor1[1] = count1 / size
    cursor2[1] = count2 / size
    data1[i] = cursor1
    data2[i] = cursor2

# Create an empty graph
myGraph = ot.Graph("Some curves", "x1", "x2", True, "bottomright")

# Create the first staircase
myStaircase1 = ot.Staircase(data1, "blue", "solid", "s", "")

myStaircase1b = ot.Staircase(myStaircase1)
myStaircase1b.setPattern("S")
myStaircase1b.setColor("green")
myStaircase1b.setLineStyle("dashed")
myStaircase1b.setLegend("eCDF1b, pat=S")

# Then, draw it
myGraph.add(myStaircase1)
myGraph.add(myStaircase1b)

# Check that the correct files have been generated by computing their
# checksum

# Create the second staircase
myStaircase2 = ot.Staircase(data2, "red", "dashed", "S", "eCDF2, pat=S")

# Add it to the graph and draw everything
myGraph.add(myStaircase2)