data.table相同组的非连续记录行之间的差异

时间:2017-08-31 14:21:59

标签: r data.table run-length-encoding

我需要计算变量的非连续记录之间的差异,按另一个分组。也就是说,我想在运行中获取变量的最后一个值,并从下一次运行中的第一个值中减去它(如果有的话)。

我知道我可以使用rleidshift计算连续行的差异,但这次我需要摆脱它们。

示例数据

dput(iris)

structure(list(Sepal.Length = c(4.4, 6.3, 4.6, 5.8, 6.4, 6.5, 
4.9, 5.4, 6.4, 6.7), Sepal.Width = c(3, 2.8, 3.1, 2.7, 2.7, 3, 
3.6, 3.9, 2.8, 3.1), Petal.Length = c(1.3, 5.1, 1.5, 4.1, 5.3, 
5.5, 1.4, 1.7, 5.6, 4.7), Petal.Width = c(0.2, 1.5, 0.2, 1, 1.9, 
1.8, 0.1, 0.4, 2.1, 1.5), Species = c("setosa", "virginica", 
"setosa", "versicolor", "virginica", "virginica", "setosa", "setosa", 
"virginica", "versicolor")), .Names = c("Sepal.Length", "Sepal.Width", 
"Petal.Length", "Petal.Width", "Species"), row.names = c(NA, 
-10L), class = c("data.table", "data.frame"))

library(data.table)
setDT(iris, key = "Sepal.Width")

我喜欢

 iris[, diff(Petal.Width), by = .(Species, !rleid(Species))]

(当然这不起作用!)是我所需要的,但却无法想到实现它。

预期结果(diff ing Petal.Width):

      Species   V1
1: versicolor  0.5
2:  virginica -0.3
3:     setosa  0.0
4:     setosa -0.1

(我实现了iris[, diff(Petal.Width), by = .(Species)],然后手工挑选.Last.Value[, c(1, 4, 5, 6)]

2 个答案:

答案 0 :(得分:2)

嗯,有

iris[, .(first(Petal.Width), last(Petal.Width)), by=.(Species, rleid(Species))][, 
  tail(V1 - shift(V2), -1), by=Species]

      Species   V1
1: versicolor  0.5
2:  virginica -0.3
3:     setosa  0.0
4:     setosa -0.1

或者...

iris[, Petal.Width[c(1L, .N)], by=.(Species, rleid(Species))][, {
  v = V1[-c(1L, .N)]
  v[c(TRUE,FALSE)] - v[c(FALSE,TRUE)]
}, by=Species]

      Species   V1
1: versicolor -0.5
2:  virginica  0.3
3:     setosa  0.0
4:     setosa  0.1

答案 1 :(得分:0)

在尝试了几件事后,我带来了这个hacky解决方案。我想有些东西会更清洁":

iris[, .(Species, Petal.Width, rl= rleid(Species))][, .(pd= ifelse(diff(rl)>0, diff(Petal.Width), NA)), by = Species][!is.na(pd),]

如果有一个以清洁方式实现这一目标的功能,我很欣赏指针。