R-программирование: как вычислить разницу между двумя ячейками во фрейме данных и сохранить их в новом столбце - PullRequest
5 голосов
/ 23 февраля 2012

Пытаюсь выучить R и застрял на примере автокорреляции.Я хочу регрессировать разницу в х с разницей в у.У меня есть x и y во фрейме данных, и я хотел бы, чтобы разница x2 - x1 была сохранена в новом столбце, скажем, dx.Я не знаю, как это сделать.

что у меня есть:

data1

x   y
5   3
8   9
3   1
1   5
.   .
.   .
.   .

что я хотел бы получить:

data1.dif

x   y   dx   dy
5   3   NA   NA
8   9    3    6
3   1   -5   -8
1   5   -2    4
.   .    .    .
.   .    .    .

Ответы [ 2 ]

8 голосов
/ 23 февраля 2012

Используйте diff с transform:

dat <- read.table(text="x   y
5   3
8   9
3   1
1   5", header=T)


transform(dat, dx=c(NA, diff(x)), dy=c(NA, diff(y)))

Уступая:

  x y dx dy
1 5 3 NA NA
2 8 9  3  6
3 3 1 -5 -8
4 1 5 -2  4

и как ог dplyr :

library(dplyr)

dat %>%
    mutate(dx=c(NA, diff(x)), dy=c(NA, diff(y)))
5 голосов
/ 23 февраля 2012

Используйте diff и прикрепите NA к началу результирующих векторов.

например.

data1 <- read.table(text='  x y
1 5 3
2 8 9
3 3 1
4 1 5')

# diff calculates the difference between consecutive pairs of 
#  vector elements
diff(data1$x)
[1]  3 -5 -2

# apply diff to each column of data1, bind an NA row to the beginning,
#  and bind the resulting columns to the original df
data1.dif <- cbind(data1, rbind(NA, apply(data1, 2, diff)))
names(data1.dif) <- c('x', 'y', 'dx', 'dy')

data1.dif
  x y dx dy
1 5 3 NA NA
2 8 9  3  6
3 3 1 -5 -8
4 1 5 -2  4
...