Doomsday fuel foobar

import numpy as np

# Returns indexes of active & terminal states
def detect_states(matrix):
    active, terminal = [], []
    for rowN, row in enumerate(matrix):
        (active if sum(row) else terminal).append(rowN)
    return(active,terminal)

# Convert elements of array in simplest form
def simplest_form(B):
    B = B.round().astype(int).A1                   # np.matrix --> np.array
    gcd = np.gcd.reduce(B)
    B = np.append(B, B.sum())                      # append the common denom
    return (B / gcd).astype(int)

# Finds solution by calculating Absorbing probabilities
def solution(m):
    active, terminal = detect_states(m)
    if 0 in terminal:                              # special case when s0 is terminal
        return [1] + [0]*len(terminal[1:]) + [1]
    m = np.matrix(m, dtype=float)[active, :]       # list --> np.matrix (active states only)
    comm_denom = np.prod(m.sum(1))                 # product of sum of all active rows (used later)
    P = m / m.sum(1)                               # divide by sum of row to convert to probability matrix
    Q, R = P[:, active], P[:, terminal]            # separate Q & R
    I = np.identity(len(Q))
    N = (I - Q) ** (-1)                            # calc fundamental matrix
    B = N[0] * R * comm_denom / np.linalg.det(N)   # get absorbing probs & get them close to some integer
    return simplest_form(B)

Source: gist.github.com

Similar codes

0 vote304 views1 answer
Pandas unqiue value each column
Python
0 vote293 views1 answer
Python regex replace point with underscore
Python
0 vote301 views1 answer
Comment arrêter l'exécution d'un programme python
Python
1 vote295 views2 answers
Read header of csv file python
Python
30 votes311 views3 answers
Iterating index array python
Python
0 vote276 views1 answer
In <lambda> tk.Button(self.frame, text = text, command= lambda:self.new_window(number, _class)).grid(padx= 5, pady= 5, sticky= N + E + W + S) File "C:\Users\CSB\Desktop\python program\triple_window.py", line 50, in new_window
Python
11 votes294 views3 answers
Practicing python
Python
0 vote294 views1 answer
Pickle.load from gpu device to cpu
Python
0 vote323 views1 answer
Python start process in background and get pid
Python
5 votes327 views1 answer
Use time now to caculate execution time python
Python