R:通过对数据表进行分组，获取每组的第一行及其对应的行号

Question

我想知道使用数据表，如何使用“特殊符号”的组合来获取第一行和每个组的数据集中相应的行号(在引用文献中)？

例如:

Library(data.table)  
copy(mtcars) -> mt
setDT(mt) -> mt

获取cyl每组的第一行:

 mt[, .SD[1], by = cyl]

    cyl  mpg disp  hp drat   wt  qsec vs am gear carb
1:   6 21.0  160 110 3.90 2.62 16.46  0  1    4    4
2:   4 22.8  108  93 3.85 2.32 18.61  1  1    4    1
3:   8 18.7  360 175 3.15 3.44 17.02  0  0    3    2

根据cyl获取每组对应的mt中的行号:

mt[, .I[1], by = cyl]
   cyl V1
1:   6  1
2:   4  3
3:   8  5

预期输出:

    cyl  mpg disp  hp drat   wt  qsec vs am gear carb   row_N
1:   6 21.0  160 110 3.90 2.62 16.46  0  1    4    4    1
2:   4 22.8  108  93 3.85 2.32 18.61  1  1    4    1    3
3:   8 18.7  360 175 3.15 3.44 17.02  0  0    3    2    5

我尝试了以下方法，但没有成功:

mt[, .SD[1], by= cyl][mt[, .I[1], by=cyl]]
mt[, .SD[1], by= cyl][mt[, `:=` (row_N = .I[1], by=cyl)]]

非常感谢任何解释帮助。

Answer 1

一种方法是使用 cbind 将列组合在一起。

mt[, cbind(.SD[1], row_N=.I[1]), by = cyl]
   cyl  mpg disp  hp drat   wt  qsec vs am gear carb row_N
1:   6 21.0  160 110 3.90 2.62 16.46  0  1    4    4     1
2:   4 22.8  108  93 3.85 2.32 18.61  1  1    4    1     3
3:   8 18.7  360 175 3.15 3.44 17.02  0  0    3    2     5