Gradient Function Python Numpy