Если можно ускорить код одной строкой, почему это не сделать ).jit - Numba