simplity construction of sparse (transition) matrix

Question

I am constructing a transition matrix from a n1 x n2 x ... x nN x nN array. For concreteness let N = 3, e.g.,

import numpy as np

# exam