File: t_GridLayout_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 (45 lines) | stat: -rwxr-xr-x 1,228 bytes parent folder | download
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
#!/usr/bin/env python

import openturns as ot
import sys

ot.TESTPREAMBLE()

grid = ot.GridLayout(2, 3)
palette = ot.Drawable.BuildDefaultPalette(10)
for j in range(grid.getNbColumns()):
    alpha = 1.0 + j
    pdf_curve = ot.WeibullMin(1.0, alpha, 0.0).drawPDF()
    cdf_curve = ot.WeibullMin(1.0, alpha, 0.0).drawCDF()
    pdf_curve.setColors([palette[j]])
    cdf_curve.setColors([palette[j]])
    legends = [f"alpha={alpha}"]
    pdf_curve.setLegends(legends)
    cdf_curve.setLegends(legends)
    grid.setGraph(0, j, pdf_curve)
    grid.setGraph(1, j, cdf_curve)

assert grid.getNbColumns() == 3
assert grid.getNbRows() == 2
assert grid.getGraph(0, 0).getDrawable(0).getImplementation().getClassName() == "Curve"

grid2 = ot.GridLayout(2, 3)
for j in range(grid2.getNbColumns()):
    sample = ot.WeibullMin(1.0, alpha, 0.0).getSample(10)
    sample.stack(ot.Sample(10, 1))
    graph = ot.Graph()
    cloud = ot.Cloud(sample)
    cloud.setColor(palette[j])
    graph.add(cloud)
    grid2.setGraph(0, j, graph)
    grid2.setGraph(1, j, graph)

grid.add(grid2)

grid.setLegendPosition("upper left")
grid.setLegendCorner([0.0, 0.5])

if len(sys.argv) > 1:
    from openturns.viewer import View

    View(grid).save("grid.png")