r语言 - 使用用户定义的"for loop"函数构造数据框

Question

我正在尝试基于用户定义的函数生成数据框。我的问题是在输出中只填充了第一行。这是我正在使用的函数的示例：

df <- data.frame(cs=rep(c("T1","T2","T3","T4"),each=16),yr=rep(c(1:4), times = 4, each = 4))
sp.df <- data.frame(matrix(sample.int(100, size = 20*64,replace=T), ncol = 20, nrow = 64))
myfunc<-function(X,  system, Title)
{
  for(i in 1:4){
    Col_T <- data.frame(matrix(NA, ncol = length(X), nrow = 4))
    Col_T[i,] <- colSums(X[which(df$yr==i & df$cs==system),])
    return (Col_T)}}
myfunc(X=sp.df, system="T1", Title="T1")

我欢迎任何解决此问题的建议。谢谢。

Answer 1

该函数有两个问题：

1. 您正在覆盖Col_T，所有 NA 作为 for 循环中的第一个语句。
2. 您将从 for 循环中的函数返回。

重写如下：

myfunc <- function(X, system, Title ) {
    Col_T <- data.frame(matrix(NA, ncol=length(X), nrow=4 ));
    for (i in 1:4)
        Col_T[i,] <- colSums(X[which(df$yr==i & df$cs==system),]);
    return(Col_T);
};