R применить результат несовместим

Question

Я вызываю apply, чтобы применить функцию к каждой строке моего кадра данных, однако я получаю некоторые странные результаты. Когда я впервые запускаю apply (прогон # 1), только подмножество строк дает ожидаемый результат. После выполнения apply во второй раз (запуск # 2) некоторые из значений, которые были изначально неверны, являются правильными. Это последовательно, в котором строки являются неправильными после запуска # 1.

assign_id() ищет идентификатор, расположенный в first вместе с девятью другими столбцами внутри кадра данных, возвращая целое число, соответствующее соответствующему столбцу.

assign_id <- function(row) {
  if(is.na(row['first'])) {
    return(NULL)
  }
  else if(row['first'] %in% c('none')) {
    return(0)
  }
  else if(row['first'] %in% as.character(row['one'])){
    return(1)
  }
  else if(row['first'] %in% as.character(row['two'])){
    return(2)
  }
  else if(row['first'] %in% as.character(row['three'])){
    return(3)
  }
  else if(row['first'] %in% as.character(row['four'])){
    return(4)
  }
  else if(row['first'] %in% as.character(row['five'])){
    return(5)
  }
  else if(row['first'] %in% as.character(row['six'])){
    return(6)
  }
  else if(row['first'] %in% as.character(row['seven'])){
    return(7)
  }
  else if(row['first'] %in% as.character(row['eight'])){
    return(8)
  }
  else if(row['first'] %in% as.character(row['nine'])){
    return(9)
  } else {
    return(11)
  }
}

df <- read.csv('df.csv')

# Run #1
df$id <- apply(df, 1, assign_id)
# All 'id' fields return 11
df[df$first %in% 55627, c('id', 'first', 'six')] 

> head(df[df$first %in% 55627, c('id', 'first', 'six')])
     id first    six
414  11 55627  55627
529  11 55627 118950
791  11 55627  55627
1570 11 55627 118950
1832 11 55627 118950
2116 11 55627 118950

# Run #2
df$id <- apply(df, 1, assign_id)
# All 'id' fields return the correct integer
df[df$first %in% 55627, c('id', 'first', 'six')] 

> head(df[df$first %in% 55627, c('id', 'first', 'six')])
     id first    six
414   6 55627  55627
529   5 55627 118950
791   6 55627  55627
1570  8 55627 118950
1832  5 55627 118950
2116  5 55627 118950

Данные расположены здесь

Answer 1

С небольшой помощью друзей я придумал более простое решение base-R:

df$id<-unlist(apply(df,1,function(x)
  ifelse(x["first"]=="none",0, which(as.integer(x["first"])==as.integer(x[2:10])))))

См. Ответы там, чтобы объяснить, почему apply был проблематичным - вкратце, он преобразовал все ваши данные в символ, но затем дополнил их так, что сравнения не увенчались успехом.

В связанной заметке, когда вы read.csv, возможно, вы захотите добавить stringsAsFactors=FALSE, чтобы избежать превращения столбца first в фактор.