Соединить функцию внутри mutate dplyr r: вычислить вторую производную

Я хотел бы рассчитать следующую формулу на всех моих данных grouped_by(id)

$$\frac{y''}{((1+(y')^2)^{3/2}}$$

Я застреваю при прохождении функции через мутирование, потому что это дает мне неравные размеры столбцов.

Пример данных:

require(dplyr)

df<-data.frame(Time=seq(65),
               SkinTemp=rnorm(65,37,0.5),
               id=rep(1:10,c(5,4,10,6,7,8,9,8,4,4)))

Первая производная (ОК):

df <-df %>% group_by(id) %>% filter(n() > 1) %>%
  mutate(first_d = SkinTemp - lag(SkinTemp))
df<-as.data.frame(df)  #Back to data frame

Вторая производная (застряла):

drv <- function(x, y) diff(y) / diff(x) #Defined this to pass to mutate

middle_pts <- function(x) x[-1] - diff(x) / 2 

second_d <-df %>% group_by(id)  %>%
  mutate(second_d = drv(middle_pts(Time), drv(Time, SkinTemp)))

1 ответ

Решение

diff возвращает вектор длиной на единицу меньше исходного vector, Функция mutate требует, чтобы столбец возвращал ту же длину. Поэтому нам нужно добавить несколько NA или ценность выбора

drv <- function(x, y) c(NA, (diff(y) /diff(x))) 
middle_pts <- function(x) c(NA, (x[-1] - diff(x) /2))

Теперь код ОП должен работать

df %>%
   group_by(id)  %>%
   mutate(second_d = drv(middle_pts(Time), drv(Time, SkinTemp)))
Другие вопросы по тегам