Python Apply Function To Each Row Of Numpy Array