python for in range get index