python if item in list get index