Python Numpy Apply Function To Each Row