to_sparse.py

def to_sparse(p_dense):
    states = numpy.indices(numpy.shape(p_dense))
    p_sparse = {}
    for state in states.transpose():
        state = tuple(state)
        probability = p_dense[state]
        if probability > 0:
            p_sparse[state] = p_dense[state]
    return p_sparse