python for in list get index