Question

数据来自我正在玩的另一个问题：

dt <- data.table(user=c(rep(3, 5), rep(4, 5)),
                 country=c(rep(1,4),rep(2,6)),
                 event=1:10, key="user")
#    user country event
#1:     3       1     1
#2:     3       1     2
#3:     3       1     3
#4:     3       1     4
#5:     3       2     5
#6:     4       2     6
#7:     4       2     7
#8:     4       2     8
#9:     4       2     9
#10:    4       2    10

这是令人惊讶的行为：

dt[user == 3, as.data.frame(table(country))]
#  country Freq
#1       1    4
#2       2    1

dt[user == 4, as.data.frame(table(country))]
#  country Freq
#1       2    5

dt[, as.data.frame(table(country)), by = user]
#   user country Freq
#1:    3       1    4
#2:    3       2    1
#3:    4       1    5
#             ^^^ - why is this 1 instead of 2?!

感谢mnel和Victor K.自然的后续行动是 - 不应该是2，即这是一个错误吗？我期待

dt[, blah, by = user]

将相同的结果返回到

rbind(dt[user == 3, blah], dt[user == 4, blah])

这种期望是否不正确？

Answer 1

惯用的data.table方法是使用.N

 dt[ , .N, by = list(user, country)]

这将更快，并且还将国家保留为与原始国家相同的班级。

Answer 2

在评论中注明mnel时，as.data.frame(table(...))会生成一个数据框，其中第一个变量是一个因子。对于user == 4，因子中只有一个级别，内部存储为1。

你想要的是因子级别，但你得到的是内部存储因子的方式（从1开始整数）。以下提供了预期结果：

> dt[, lapply(as.data.frame(table(country)), as.character), by = user]
   user country Freq
1:    3       1    4
2:    3       2    1
3:    4       2    5

<强>更新即可。关于你的第二个问题：不，我认为data.table行为是正确的。当你加入两个不同级别的因子时，在普通R中也会发生同样的事情：

> a <- factor(3:5)
> b <- factor(6:8)
> a
[1] 3 4 5
Levels: 3 4 5
> b
[1] 6 7 8
Levels: 6 7 8
> c(a,b)
[1] 1 2 3 1 2 3

data.table和表意外行为

2 个答案: