python for loop get index number