import numpy as np
import matplotlib.pyplot as plt


n = 100
p = 0.1
P = []
C = []

X = range(1,n)
for i in range(1,n):
    P.append(p*((1-p)**(i-1)))
    C.append(1-(1-p)**i)

    
plt.figure('Probability Distribution')
plt.plot(X, P)
x_axis = np.arange(1, n, 10)
y_axis = np.arange(0, max(P)+0.02, 0.02)
plt.xticks(x_axis)
plt.yticks(y_axis)
plt.xlabel('x')
plt.ylabel('Pr[X=x]')
plt.title('Geometric PDF with p=%.2f' %p)

if(1):
    plt.figure('Cumulative Distribution')
    plt.plot(X, C)
    x_axis = np.arange(1, n, 10)
    y_axis = np.arange(0, 1.1, 0.1)
    plt.xticks(x_axis)
    plt.yticks(y_axis)
    plt.xlabel('x')
    plt.ylabel('Pr[X<=x]')
    plt.title('Geometric CDF with p=%.2f' %p)
    
plt.grid()
plt.show()