How To Return An Index In Python