保持行仅对应于列中的某些值

时间:2016-03-01 14:28:07

标签: r data.table reshape reshape2

我有一个data.table,如下面的

    Sepal.Length Sepal.Width Petal.Length Petal.Width Species pg rem_imp
 1:          5.1         3.5          1.4         0.2     qwe  1       3
 2:          4.9         3.0          1.4         0.2     qwe  1       3
 3:          4.7         3.2          1.3         0.2     qwe  5       3
 4:          4.6         3.1          1.5         0.2     qwe  2       3
 5:          5.0         3.6          1.4         0.2     qwe  2       3
 6:          5.4         3.9          1.7         0.4     qwe  2       3
 7:          4.6         3.4          1.4         0.3     qwe  3       3
 8:          5.0         3.4          1.5         0.2     qwe  3       3
 9:          4.4         2.9          1.4         0.2     qwe  3       3
10:          4.9         3.1          1.5         0.1     qwe  3       3
11:          5.4         3.7          1.5         0.2  setosa 11       2
12:          4.8         3.4          1.6         0.2  setosa 11       2
13:          4.8         3.0          1.4         0.1  setosa 15       2
14:          4.3         3.0          1.1         0.1  setosa 15       2
15:          5.8         4.0          1.2         0.2  setosa 13       2

structure(list(Sepal.Length = c(5.1, 4.9, 4.7, 4.6, 5, 5.4, 4.6, 
5, 4.4, 4.9, 5.4, 4.8, 4.8, 4.3, 5.8), Sepal.Width = c(3.5, 3, 
3.2, 3.1, 3.6, 3.9, 3.4, 3.4, 2.9, 3.1, 3.7, 3.4, 3, 3, 4), Petal.Length = c(1.4, 
1.4, 1.3, 1.5, 1.4, 1.7, 1.4, 1.5, 1.4, 1.5, 1.5, 1.6, 1.4, 1.1, 
1.2), Petal.Width = c(0.2, 0.2, 0.2, 0.2, 0.2, 0.4, 0.3, 0.2, 
0.2, 0.1, 0.2, 0.2, 0.1, 0.1, 0.2), Species = c("qwe", "qwe", 
"qwe", "qwe", "qwe", "qwe", "qwe", "qwe", "qwe", "qwe", "setosa", 
"setosa", "setosa", "setosa", "setosa"), pg = c(1, 1, 5, 2, 2, 
2, 3, 3, 3, 3, 11, 11, 15, 15, 13), rem_imp = c(3, 3, 3, 3, 3, 
3, 3, 3, 3, 3, 2, 2, 2, 2, 2)), .Names = c("Sepal.Length", "Sepal.Width", 
"Petal.Length", "Petal.Width", "Species", "pg", "rem_imp"), row.names = c(NA, 
-15L), class = c("data.table", "data.frame"), .internal.selfref = <pointer: 0x0000000011eb0788>)

我希望仅为列rem_imp的每个值保留与列pg的第一个Species值对应的行。例如,For qwe我想只保留列pg的前3个(rem_imp列值)值。对于setosa,我只希望保留列pg的前2个值。

编辑:(详细解释我想要的内容)

我想要的是为每个rem_imp保留与pg的前Species个唯一值对应的行。例如,对于qwe,我希望保留与pg的前3个唯一值对应的行。 pg qwe的前3个唯一值为1,5和2,因此我们保留1到6行。setosa pg的前2个唯一值为11和25,所以我们为setosa保留11到14行。

输出如下所示

     Sepal.Length Sepal.Width Petal.Length Petal.Width Species pg rem_imp
 1:          5.1         3.5          1.4         0.2     qwe  1       3
 2:          4.9         3.0          1.4         0.2     qwe  1       3
 3:          4.7         3.2          1.3         0.2     qwe  5       3
 4:          4.6         3.1          1.5         0.2     qwe  2       3
 5:          5.0         3.6          1.4         0.2     qwe  2       3
 6:          5.4         3.9          1.7         0.4     qwe  2       3
 7:          5.4         3.7          1.5         0.2  setosa 11       2
 8:          4.8         3.4          1.6         0.2  setosa 11       2
 9:          4.8         3.0          1.4         0.1  setosa 15       2
10:          4.3         3.0          1.1         0.1  setosa 15       2

我确信这可以在一个声明中完成,但到目前为止我无法成功完成。 我正在寻找一个data.table语法解决方案(基本上是一行解决方案)。我不想运行循环或其他东西。我该怎么办?

1 个答案:

答案 0 :(得分:2)

这是一个选项:

dt[dt[, cumsum(!duplicated(pg)) <= rem_imp[1L], by = Species]$V1]
#    Sepal.Length Sepal.Width Petal.Length Petal.Width Species pg rem_imp
# 1:          5.1         3.5          1.4         0.2     qwe  1       3
# 2:          4.9         3.0          1.4         0.2     qwe  1       3
# 3:          4.7         3.2          1.3         0.2     qwe  5       3
# 4:          4.6         3.1          1.5         0.2     qwe  2       3
# 5:          5.0         3.6          1.4         0.2     qwe  2       3
# 6:          5.4         3.9          1.7         0.4     qwe  2       3
# 7:          5.4         3.7          1.5         0.2  setosa 11       2
# 8:          4.8         3.4          1.6         0.2  setosa 11       2
# 9:          4.8         3.0          1.4         0.1  setosa 15       2
#10:          4.3         3.0          1.1         0.1  setosa 15       2