[英]Fill in Missing Weekly Data Points in R as 0 for Each ID
我有这种形状的数据:
> head(posts)
id week_number num_posts
1 UKL1.1 1 4
2 UKL1.1 6 9
3 UKL1.2 1 2
4 UKL1.3 1 8
5 UKL1.3 2 7
6 UKL1.3 3 3
我想使每个id
在每个week_number
(1,2,3,4,5,6)中都有一行,如果数据中还没有那个week_number
,那么posts
应该= 0
我已经看到了这个用包做zoo
与真正的时间序列数据,而无需创建一个适当的POSIXct
或Date
的版本week_number
,并使用该包装是有没有办法直接做到这一点?
这是使用data.table
的方法。
library(data.table)
setDT(posts) # convert posts to a data.table
all.wks <- posts[,list(week_number=min(week_number):max(week_number)),by=id]
setkey(posts,id,week_number) # index on id and week number
setkey(all.wks,id,week_number) # index on id and week number
result <- posts[all.wks] # data.table join is very fast
result[is.na(num_posts),num_posts:=0] # convert NA to 0
result
# id week_number num_posts
# 1: UKL1.1 1 4
# 2: UKL1.1 2 0
# 3: UKL1.1 3 0
# 4: UKL1.1 4 0
# 5: UKL1.1 5 0
# 6: UKL1.1 6 9
# 7: UKL1.2 1 2
# 8: UKL1.3 1 8
# 9: UKL1.3 2 7
# 10: UKL1.3 3 3
其他方式:
my_fun <- function(x) {
weeks = with(x, min(week_number):max(week_number))
posts = with(x, num_posts[match(weeks, week_number)])
list(week_number=weeks, num_posts=posts)
}
setDT(posts)[, my_fun(.SD), by=id]
.SD
表示数据的子集; 它包含对应于规定的各组的子集的数据by
,与不包括分组列=所有的列id
。
然后,您可以如上所述替换NA
。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.