Return Function In Numpy