如何找到随机选择的数据帧行的补码？ [复制]

Question

我有一个名为 data 的数据集，我将其拆分为 2 个新数据集，分别称为 test 和 train。

我希望拆分是随机的，没有替换。

使用下面的代码，我得到 train 是一个包含 35 个元素的新数据框：

rows_in_test <-  35  # number of rows to randomly select 
rows_in_train <- nrow(data) - rows_in_test 
train <- data[sample(nrow(data), rows_in_test), ] 

R 中是否有一种很好的方法可以将train 的补码分配给一个名为test 的新数据集？我在想一定有这个功能？

Answer 1

myData<-data.frame(a=c(1:20), b=c(101:120))
set.seed(123)#to be able to replicate random sampling later
trainRows<-runif(nrow(myData))>0.25 #randomly put aside 25% of the data
train<-myData[trainRows,]#has 13 rows
test<-myData[!trainRows,]#has 7 rows

#following method to select a fixed no. of samples - in this case selecting 5 rows
testRows2<-sort(sample(c(1:nrow(myData)), 5, replace=F))

train2<-myData[-testRows2, ]
test2<-myData[testRows2, ]