使用用户定义的“ for循环”功能构造数据帧

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. 您将用所有NA覆盖Col_T作为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);
};