Simple Markov Chain visualization module in Python. Only requires matplotlib and numpy to work.
The current version works with 2 to 4 states.
- matplotlib
- numpy
Copy the files src/node.py and src/markovchain.py in your script directory. Then
from markovchain import MarkovChain
P = np.array([[0.8, 0.2], [0.1, 0.9]]) # Transition matrix
mc = MarkovChain(P, ['1', '2'])
mc.draw("../img/markov-chain-two-states.png")
P = np.array([
[0.8, 0.1, 0.1],
[0.1, 0.7, 0.2],
[0.1, 0.7, 0.2],
])
mc = MarkovChain(P, ['A', 'B', 'C'])
mc.draw("../img/markov-chain-three-states.png")
P = np.array([
[0.8, 0.1, 0.1, 0.0],
[0.1, 0.7, 0.0, 0.2],
[0.1, 0.0, 0.7, 0.2],
[0.1, 0.0, 0.7, 0.2]
])
mc = MarkovChain(P, ['1', '2', '3', '4'])
mc.draw("../img/markov-chain-four-states.png")
Link to my blog.
This project is licensed under the GPL V3 licence.