Pagerank - irdm sheet 11 problem 1

pagerank.y

import numpy as np

def iterate(alpha, n, P, pi):
    e = np.array([[1], [1], [1], [1], [1], [1]])

    return (alpha * e)/n + (1-alpha) * np.matmul(np.transpose(P), pi)

if __name__ == "__main__":
    P = np.array([
        [0, 0.5, 0, 0.5, 0, 0],
        [0, 0, 0, 0.5, 0, 0.5],
        [0, 0.5, 0, 0, 0, 0.5],
        [0, 0, 1, 0, 0, 0],
        [0, 0.5, 0, 0.5, 0, 0],
        [0, 0, 0, 0, 1, 0]
        ])
    pi = np.array([[1/6], [1/6], [1/6], [1/6], [1/6], [1/6]])
    alpha = 0.15
    n = 6

    for i in range(5):
        pi = iterate(alpha, n, P, pi)
   
    print(pi)

   # pi = iterate(alpha, n, P, pi)
   # print(pi)

    #pi = iterate(alpha, n, P, pi)
    #print(pi)