data.table 中的行号

Question

我有这个简单的任务要做，想象一下我有这张表:

   var1
1:    X
2:    X
3:    Y
4:    X
5:    Z
6:    Y

我想像这样创建一个新变量:

   var1 nth
1:    X   1
2:    X   2
3:    Y   1
4:    X   3
5:    Z   1
6:    Y   2

实际上我有很多变量很多行。所以我的实际解决方案是(效果很好):

dt[, flag := 1]
dt[, nth := frank(flag, ties.method = "random"), by = .(var1)]

但这很慢，而且我对排名并不感兴趣。(frank 必须排序和排名，我只需要一个简单的东西，即索引)。

我想出了这个解决方案，但行不通:

dt[, pos := .I, by = .(var1)]

对于这项任务，您有任何快速直接的解决方案吗？

谢谢

Answer 1

dt[, nth := row.names(.SD), by = var1]