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.NegativeBinomial(r, p)
    pdf_curve = distribution.drawPDF().getDrawable(0)
    cdf_curve = distribution.drawCDF().getDrawable(0)
    pdf_curve.setColor(palette[i])
    cdf_curve.setColor(palette[i])
    pdf_curve.setLegend("r,p={},{}".format(r, p))
    cdf_curve.setLegend("r,p={},{}".format(r, p))
    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("NegativeBinomial(r,p)")
