data.table按組分配向量

Question

假設我們有以下data.table

dt = data.table(a=letters[1:20], b = c(rep(1,3),rep(2,7),rep(3,5),rep(4,5)))

那是

    a b
 1: a 1
 2: b 1
 3: c 1
 4: d 2
 5: e 2
 6: f 2
 7: g 2
 8: h 2
 9: i 2
10: j 2
11: k 3
12: l 3
13: m 3
14: n 3
15: o 3
16: p 4
17: q 4
18: r 4
19: s 4
20: t 4

我想為每行分配一個從0到1的等級，但按b列分組。 我正在做

dt[,len:=.N,by=b][,rank:=c(0:(len-1))/(len-1),by=b][,len:=NULL]

len在那里只是為了計算等級，然后被刪除。 我得到

    a b      rank
 1: a 1 0.0000000
 2: b 1 0.5000000
 3: c 1 1.0000000
 4: d 2 0.0000000
 5: e 2 0.1666667
 6: f 2 0.3333333
 7: g 2 0.5000000
 8: h 2 0.6666667
 9: i 2 0.8333333
10: j 2 1.0000000
11: k 3 0.0000000
12: l 3 0.2500000
13: m 3 0.5000000
14: n 3 0.7500000
15: o 3 1.0000000
16: p 4 0.0000000
17: q 4 0.2500000
18: r 4 0.5000000
19: s 4 0.7500000
20: t 4 1.0000000

這正是我想要的。 問題是我也得到這個

   Warning messages:
 1: In base::":"(from, to) :
  numerical expression has 3 elements: only the first used
 2: In base::":"(from, to) :
  numerical expression has 7 elements: only the first used
 3: In base::":"(from, to) :
  numerical expression has 5 elements: only the first used
 4: In base::":"(from, to) :
  numerical expression has 5 elements: only the first used

我想不理會它們，當數據很小並且我可以目視檢查結果時，這很好。 但是由於我的data.table具有數千行，因此我想確保這些警告實際上是無害的。

你怎么看？ 或者，等效地，我通過分組data.table分配“向量”的方法是否被允許？ 有其他選擇嗎？

謝謝

Answer 1

您將從以下代碼部分獲得警告： 0:(len-1) 。 : ， len-1的第二個參數是長度為.N的向量，但:想要長度為1的向量。您可以使用(1:2):(2:3)或seq_len(2):seq_len(2)重新創建警告seq_len(2):seq_len(2) 。

下面將在一行中計算您想要的內容，而不會發出警告：

dt[, rank := (seq_len(.N) - 1) / (.N - 1), by=b]
dt
    a b      rank
 1: a 1 0.0000000
 2: b 1 0.5000000
 3: c 1 1.0000000
 4: d 2 0.0000000
 5: e 2 0.1666667
 6: f 2 0.3333333
 7: g 2 0.5000000
 8: h 2 0.6666667
 9: i 2 0.8333333
10: j 2 1.0000000
11: k 3 0.0000000
12: l 3 0.2500000
13: m 3 0.5000000
14: n 3 0.7500000
15: o 3 1.0000000
16: p 4 0.0000000
17: q 4 0.2500000
18: r 4 0.5000000
19: s 4 0.7500000
20: t 4 1.0000000