import numpy as np
import matplotlib.pyplot as plt


p = 0.3
P = []
C = []

X = [0, 1]
P = [1-p, p]
C = [1-p, 1]

plt.figure('Probability Distribution')
plt.bar(X, P, 0.6)
x_axis = range(len(X))
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('Bernoulli PDF with p=%.2f' %p)

if(1):
    plt.figure('Cumulative Distribution')
    plt.bar(X, C)
    x_axis = np.arange(len(X))
    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('Bernoulli CDF with p=%.2f' %p)
plt.grid()
plt.show()