Or you can use numba and with pure python syntax and get really fast.