How To Return Index In Python