Apply Lambda Function To Numpy Python