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
|
import openturns as ot
from matplotlib import pyplot as plt
from openturns.viewer import View
pdf_graph = ot.Graph("PDF graph", "x", "PDF", True, "upper right")
cdf_graph = ot.Graph("CDF graph", "x", "CDF", True, "lower right")
palette = ot.Drawable.BuildDefaultPalette(10)
for i, v in enumerate([(2, 0.83), (4, 0.66), (20, 0.33)]):
r, p = v
distribution = ot.Polya(r, p)
pdf_curve = distribution.drawPDF().getDrawable(0)
cdf_curve = distribution.drawCDF().getDrawable(0)
pdf_curve.setColor(palette[i])
cdf_curve.setColor(palette[i])
legend = f"r={r}, p={p}"
pdf_curve.setLegend(legend)
cdf_curve.setLegend(legend)
pdf_graph.add(pdf_curve)
cdf_graph.add(cdf_curve)
fig = plt.figure(figsize=(10, 4))
pdf_axis = fig.add_subplot(121)
cdf_axis = fig.add_subplot(122)
View(pdf_graph, figure=fig, axes=[pdf_axis], add_legend=True)
View(cdf_graph, figure=fig, axes=[cdf_axis], add_legend=True)
fig.suptitle("Polya(r,p)")
|
