First hitting time python

from numpy import *

hit_idx = (0,4)

# Define a graph by edge list
edges = [[0,1],[1,2],[2,3],[2,4]]

# Create adj. matrix
A = zeros((5,5))
A[zip(*edges)] = 1
# Undirected condition
A += A.T

# Make the final state an absorbing condition
A[hit_idx[1],:] = 0
A[hit_idx[1],hit_idx[1]] = 1

# Make a proper Markov matrix by row normalizing
A = (A.T/A.sum(axis=1)).T

B = A.copy()
Z = []
for n in xrange(100):
    Z.append( B[hit_idx] )
    B = dot(B,A)

from pylab import *
plot(Z)
xlabel("steps")
ylabel("hit probability")
show()

Source: stackoverflow.com

Similar codes

1 vote132 views1 answer
Display max number of columns pandas
Python
9 votes117 views1 answer
How to make it so a discord bot messages in a certain channel python
Python
0 vote129 views1 answer
Write dict to json file with special characters
Python
0 vote150 views1 answer
Create column for year in dataframe python
Python
1 vote143 views1 answer
Pandas replace colomns location
Python
1 vote139 views2 answers
Pygame window at center
Python
1 vote156 views1 answer
Last 24 hour python datetime
Python
0 vote133 views1 answer
Python hide print output
Python
0 vote136 views1 answer
Structural pattern matching python
Python
0 vote121 views1 answer
How to compare current date to future date pythono
Python