Python Function Return Variable Number Definition Geometry