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 vote247 views1 answer
Sort array python by column
Python
0 vote298 views1 answer
How to get rid of all null values in array python
Python
16 votes257 views6 answers
Turtle with python
Python
1 vote237 views1 answer
Deleting an object in python
Python
17 votes203 views5 answers
Read data from csv python
Python
2 votes278 views1 answer
Add a list in python
Python
0 vote256 views1 answer
Ocaml returns the last element of a list
Python
0 vote269 views1 answer
Seaborn heatmap spearman correlation coefficient
Python
0 vote241 views1 answer
Replace \u00a0
Python
1 vote273 views1 answer
How to retrieve dictionary values in python by index
Python