时间戳

首页 > 解决方案 > 来自 data.table 的样本

r - 来自 data.table 的样本

问题描述

我有一些 data.table 我想从中选择一个随机子集，但仅用于某些操作。

假设数据是

dat <- data.table(id=1:100, group=sample(1:20,100, replace=TRUE), a=runif(100), b=rnorm(100))

我想做两件事：

  1. 计算每组 id 的数量
  2. 从每个组中随机选择一个 id 并将其值记录a在b

我可以关注How do you extract a few random rows from a data.table on the fly并选择

dat[n=.N, a=a[sample(.N,1)], b=b[sample(.N,1)], group]

但我担心，这将a彼此b独立选择。有没有办法选择相同的？

标签： rdata.tablesubset

解决方案

第1部分

如果要计算唯一 id 的数量并且某些 id 在组内重复

dat[, .(n_ids = uniqueN(id)), group]

如果 id 不在组内重复，或者您不想在唯一的基础上计算它们

dat[, .(n_ids = .N), group]

第2部分

如果 id 在组内重复，并且您希望返回每个组中随机选择的 id 的所有行

dat[dat[, .(id = sample(id, 1)), group], on = .(id, group)]

如果 ids 不重复，或者你只想要每组一行

dat[dat[, sample(.I, 1), group]$V1]

感谢 Frank 的评论，您还可以在一行中为上面的第 1 部分和第 2 部分执行第二个选项。这会返回类似的行，dat[dat[, sample(.I, 1), group]$V1]但还会添加一列N显示 id 的数量（假设等于组中的行数）

dat[sample(.N), c(.SD[1], .N), keyby=group]

推荐阅读