我有一个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语法解决方案(基本上是一行解决方案)。我不想运行循环或其他东西。我该怎么办?
答案 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