Python Function Return Variable Number Definition In Math