在R中创建一个新变量

时间:2016-08-30 20:00:33

标签: r

数据集包含四列(id,x1,x2和y1)。请注意,有一些多条记录(按ID)。

以下是数据:

id <- c(1,  1,  1,  1,  2,  3,  3,  3,  4,  4,  4,  4,  5,  5, 5, 6)    
x1 <- c("a","b","c","a","a","a","c","c","b", "e", "w", "r", "b", "c", "w", "r")
x2 <- c(0.12,   0.76,   0.08,   0.11,   0.80,   0.24,   0.19,   0.07,   0.70,   0.64,   0.97,   0.04,   0.40,   0.67,   0.25, 0.01)
y1 <- c(1132,   1464,   454,    1479,   167,    335,    280,    391,    973,    1343,   777,    1333,   293,    694,    76, 114)
mdat <- data.frame(id, x1, x2, y1)

我想创建一个新列(让我们称之为y2)。 ys定义为

对于相同的id,

y2(i)= y1(i-1)。对于具有onlu一个id的数据,不是那个,那么y2 = NA。

这是输出:

id  x1  x2       y1     y2
1   a   0.12    1132    
1   b   0.76    1464    1132
1   c   0.08    454     1464
1   a   0.11    1479    454
2   a   0.8     167 
3   a   0.24    335 
3   c   0.19    280     335
3   c   0.07    391     280
4   b   0.7     973 
4   e   0.64    1343    973
4   w   0.97    777     1343
4   r   0.04    1333    777
5   b   0.4     293 
5   c   0.67    694     293
5   w   0.25    76      694
6   r   0.01    114 

2 个答案:

答案 0 :(得分:2)

您可以考虑使用lag包中的dplyr功能

来考虑替代方案
> mdat$y2 <- unlist(tapply(mdat$y1, mdat$id, lag, 1))
> mdat
   id x1   x2   y1   y2
1   1  a 0.12 1132   NA
2   1  b 0.76 1464 1132
3   1  c 0.08  454 1464
4   1  a 0.11 1479  454
5   2  a 0.80  167   NA
6   3  a 0.24  335   NA
7   3  c 0.19  280  335
8   3  c 0.07  391  280
9   4  b 0.70  973   NA
10  4  e 0.64 1343  973
11  4  w 0.97  777 1343
12  4  r 0.04 1333  777
13  5  b 0.40  293   NA
14  5  c 0.67  694  293
15  5  w 0.25   76  694
16  6  r 0.01  114   NA

答案 1 :(得分:1)

dplyr

的解决方案
library(dplyr)

mdat %>%
  group_by(id) %>%
  mutate(y2 = y1 - c(NA,y1[-length(y1)]))