如何在 R 中划分一组值（向量）

Question

我正在使用 R 进行编程。我有一个包含 1000 个值的向量。现在假设我想将这 1000 个值随机划分为两个新集合，一组包含 400 个值，另一个包含 600 个值。我该怎么做？我想过做这样的事情......

firstset <- sample(mydata, size=400)

但这不会对数据进行分区（换句话说，我仍然不知道将哪 600 个值放入另一组中）。我还考虑过从 1 到 400 循环，一次随机删除 1 个值并将其放入 firstset 中。这将正确分区数据，但我不清楚如何实现这一点。另外，我被告知要尽可能避免 R 中的 for 循环。

有什么想法吗？

Answer 1

您可以对它们的位置进行采样，而不是对值进行采样。

positions <- sample(length(mydata), size=400)  # ucfagls' suggestion
firstset <- mydata[positions]
secondset <- mydata[-positions]

编辑：ucfagls的建议会更有效（特别是对于较大的向量），因为它避免了在R中分配位置向量。

Answer 2

只需随机化 mydata 并取前 400 个，然后取最后 600 个。

mydata <- sample(mydata)
firstset <- mydata[1:400]
secondset <- mydata[401:1000]

Answer 3

如果 mydata 确实是一个向量，则一种选择是：

split(mydata, sample(c(rep("group1", 600), rep("group2", 400))))