Python Discard Return Value Based On Two