Trace Function In Numpy