python if value in list get index