python get index if item in list