Неверные результаты из числового сравнения в R

Question

Я хочу выполнить эту простую категоризацию в кадре данных в R. Категории - 1,2,3,4 и -1. Я хочу добавить результаты этого вычисления в новый столбец в фрейме данных. «oldCol» - это имя столбца, уже находящегося в фрейме данных.

DF$newCol <- apply(DF, 1, function(row) { 
    if (row[["oldCol"]] > 10.0) 
    {result1 <- 4.0} 
    else if (row[["oldCol"]] > 2.0 && row[["oldCol"]] <= 10.0) 
    {result1 <- 3.0} 
    else if (row[["oldCol"]] > 0.5 && row[["oldCol"]] <= 2.0) 
    {result1 <- 2.0} 
    else if (row[["oldCol"]] > 0.0 && row[["oldCol"]] <= 0.5) 
    {result1 <- 1.0} 
    else 
    {result1 <- -1.0} 
    return(result1) 
}) 

Моя проблема: код делает новый столбец, но значения в нем неверны! С помощью этого точного кода цифры более 10 корректно классифицируются как класс 4, но все остальные строки содержат -1. Зачем? Алгоритм настолько прост, что это действительно беспокоит меня.

Кроме того, есть ли более элегантный способ сделать это?

Answer 1

Ваш код работает для меня:

DF <- data.frame(oldCol=c(-1,0.25,1,5,12)) 

DF$newCol <- apply(DF, 1, function(row) { 
    if (row[["oldCol"]] > 10.0) 
    {result1 <- 4.0} 
    else if (row[["oldCol"]] > 2.0 && row[["oldCol"]] <= 10.0) 
    {result1 <- 3.0} 
    else if (row[["oldCol"]] > 0.5 && row[["oldCol"]] <= 2.0) 
    {result1 <- 2.0} 
    else if (row[["oldCol"]] > 0.0 && row[["oldCol"]] <= 0.5) 
    {result1 <- 1.0} 
    else 
    {result1 <- -1.0} 
    return(result1) 
}) 

Результаты:

## oldCol newCol 
## 1 -1.00  -1 
## 2 0.25  1 
## 3 1.00  2 
## 4 5.00  3 
## 5 12.00  4 
## 

Одна альтернатива:

DF <- transform(DF, 
     newCol=as.numeric(as.character(cut(oldCol, 
      breaks=c(-Inf,0,0.5,2,10,Inf), 
      labels=c(-1,1:4))))) 

или:

library("plyr") 
DF <- mutate(DF, 
     tmpCol=cut(oldCol, 
      breaks=c(-Inf,0,0.5,2,10,Inf),labels=FALSE), 
     newCol=ifelse(tmpCol=="1",-1,as.numeric(tmpCol)-1)) 

Answer 2

Вот несколько проще ответ, который также принимает во внимание, что вы имеете дело с double-й годами, и поэтому будет иметь точные вопросы:

cuts = c(0, 0.5, 2, 10) + 1e-8 # example precision, pick appropriately for your problem 
           # (possibly pick a different precision for each cut point) 
DF$newCol = findInterval(DF$oldCol, cuts)