通过使用for循环求和数据框的列来创建新变量

时间:2019-04-28 09:21:09

标签: r

首先,我想使用for循环模拟20个数据集。生成这些数据集后,我想在每次迭代时向这些数据集添加一个新变量(列)。新变量是每个数据集所有列的总和。

library(bindata)
set.seed(485)
cor.mat = diag(1, nrow = 3)
for (i in 1:nrow(cor.mat)) {
  for (j in 1:ncol(cor.mat)) {
    if (i > j) cor.mat[i, j] = runif(1, 0.2, 0.7)
    cor.mat[j, i] = cor.mat[i, j]
  }
}

N=20
mydata = list()
var = list()
mydata.new = list()
for(i in 1:N){
    mydata[[i]] = rmvbin(n=10,margprob=rep(0.4,3),bincorr = cor.mat)
    var[[i]] = mydata[,1][i]+mydata[,2][i]+mydata[,3][i] # I need your help here
    mydata.new[[i]] = cbind(mydata,var) # Again I need your help here
}

我希望20个mydata.new数据集如下所示。

v1 v2 v3 var
0  1  0   1
1  0  0   1
1  1  1   3
0  0  0   0
1  0  1   2

提前谢谢!

2 个答案:

答案 0 :(得分:0)

如果我理解正确,那么您想要这样做:

for(i in 1:N){
  tmp = rmvbin(n=10,margprob=rep(0.4,3),bincorr = cor.mat)
  mydata[[i]] <- cbind(tmp, rowSums(tmp))
}

> mydata
[[1]]
      [,1] [,2] [,3] [,4]
 [1,]    0    0    0    0
 [2,]    0    1    0    1
 [3,]    0    1    0    1
 [4,]    0    0    0    0
 [5,]    0    0    1    1
 [6,]    1    1    1    3
 [7,]    0    0    0    0
 [8,]    1    1    1    3
 [9,]    0    1    1    2
[10,]    0    0    0    0
...

答案 1 :(得分:0)

这能达到预期的结果吗?

AppCenter.Start