Условная замена значений в data.frame

Я пытаюсь понять, как условно заменить значения в dataframe без использования цикла. Мой кадр данных структурирован следующим образом:

> df
          a b est
1  11.77000 2   0
2  10.90000 3   0
3  10.32000 2   0
4  10.96000 0   0
5   9.90600 0   0
6  10.70000 0   0
7  11.43000 1   0
8  11.41000 2   0
9  10.48512 4   0
10 11.19000 0   0

и вывод dput:

structure(list(a = c(11.77, 10.9, 10.32, 10.96, 9.906, 10.7, 
11.43, 11.41, 10.48512, 11.19), b = c(2, 3, 2, 0, 0, 0, 1, 2, 
4, 0), est = c(0, 0, 0, 0, 0, 0, 0, 0, 0, 0)), .Names = c("a", 
"b", "est"), row.names = c(NA, -10L), class = "data.frame")

Что я хочу сделать, это проверить значение b. Если b равно 0, я хочу установить est значение из a. Я понимаю, что df$est[df$b == 0] <- 23 установит все значения est равным 23, когда b==0. Я не понимаю, как установить est значение a, когда это условие истинно. Например:

df$est[df$b == 0] <- (df$a - 5)/2.533

дает следующее предупреждение:

Warning message:
In df$est[df$b == 0] <- (df$a - 5)/2.533 :
  number of items to replace is not a multiple of replacement length

Есть ли способ, которым я могу передать соответствующую ячейку, а не вектор?

Ответ 1

Поскольку вы условно индексируете df$est, вам также необходимо условно проиндексировать вектор замены df$a:

index <- df$b == 0
df$est[index] <- (df$a[index] - 5)/2.533

Конечно, переменная index является временной, и я использую ее, чтобы сделать код более понятным. Вы можете написать это за один шаг:

df$est[df$b == 0] <- (df$a[df$b == 0] - 5)/2.533

Для еще большей читаемости вы можете использовать within:

df <- within(df, est[b==0] <- (a[b==0]-5)/2.533)

Результаты, независимо от выбранного вами метода:

df
          a b      est
1  11.77000 2 0.000000
2  10.90000 3 0.000000
3  10.32000 2 0.000000
4  10.96000 0 2.352941
5   9.90600 0 1.936834
6  10.70000 0 2.250296
7  11.43000 1 0.000000
8  11.41000 2 0.000000
9  10.48512 4 0.000000
10 11.19000 0 2.443743

Как указывали другие, альтернативным решением в вашем примере является использование ifelse.

Ответ 2

Попробуйте data.table := operator:

DT = as.data.table(df)
DT[b==0, est := (a-5)/2.533]

Он быстрый и короткий. См. Эти связанные вопросы для получения дополнительной информации о :=:

Почему имеет значение data.table :=

Когда следует использовать оператор := в data.table

Как удалить столбцы из data.frame

Самостоятельное задание R

Ответ 3

Вот один из подходов. ifelse векторизован и он проверяет все строки для нулевых значений b и заменяет est на (a - 5)/2.53, если это так.

df <- transform(df, est = ifelse(b == 0, (a - 5)/2.53, est))

Ответ 4

R-inferno, или базовая R-документация объяснит, почему использование df $* - не лучший подход здесь. На странице справки для "[":

"Индексация по [похожа на атомные векторы и выбирает список указанных элементов). И [[и $выберите один элемент списка. Основное различие заключается в том, что $не позволяет вычислять индексы, тогда как [[делает. x $name эквивалентно x [[ "name", exact = FALSE]]. Кроме того, поведение парного совпадения [[можно контролировать с помощью точного аргумента. "

Вместо этого рекомендуется использовать нотацию [row,col]. Пример:

Rgames: foo   
         x    y z  
   [1,] 1e+00 1 0  
   [2,] 2e+00 2 0  
   [3,] 3e+00 1 0  
   [4,] 4e+00 2 0  
   [5,] 5e+00 1 0  
   [6,] 6e+00 2 0  
   [7,] 7e+00 1 0  
   [8,] 8e+00 2 0  
   [9,] 9e+00 1 0  
   [10,] 1e+01 2 0  
Rgames: foo<-as.data.frame(foo)

Rgames: foo[foo$y==2,3]<-foo[foo$y==2,1]
Rgames: foo
       x y     z
1  1e+00 1 0e+00
2  2e+00 2 2e+00
3  3e+00 1 0e+00
4  4e+00 2 4e+00
5  5e+00 1 0e+00
6  6e+00 2 6e+00
7  7e+00 1 0e+00
8  8e+00 2 8e+00
9  9e+00 1 0e+00
10 1e+01 2 1e+01