advance on tests; almost there...

[talweg.git] / data / scripts / fill_NAs.R

diff --git a/data/scripts/fill_NAs.R b/data/scripts/fill_NAs.R
index 72af4cb..8032662 100644 (file)
--- a/data/scripts/fill_NAs.R
+++ b/data/scripts/fill_NAs.R
@@ -9,21 +9,17 @@ fill_NAs = function(file)
        last_noNA_indices = rep(0, p)
        while (line <= n)
        {
-#              print(line)
                for (i in 1:p)
                {
-#                      if (is.numeric(dat[line,i]))
                        if (!is.na(dat[line,i]))
                        {
                                if (last_noNA_indices[i] < line-1)
                                {
                                        #do some completion
-                                       meanVal = ifelse(last_noNA_indices[i]>0,
-                                                                                                0.5*(dat[last_noNA_indices[i],i]+dat[line,i]),
-                                                                                                dat[line,i])
-
-                                       for (j in (last_noNA_indices[i]+1):(line-1))
-                                               dat[j,i] = meanVal
+                                       meanVal = ifelse( last_noNA_indices[i]>0,
+                                               0.5*(dat[last_noNA_indices[i],i]+dat[line,i]),
+                                               dat[line,i] )
+                                       dat[(last_noNA_indices[i]+1):(line-1),i] = meanVal
                                }
                                last_noNA_indices[i] = line
                        }
@@ -35,10 +31,7 @@ fill_NAs = function(file)
        for (i in 1:p)
        {
                if (last_noNA_indices[i] < n)
-               {
-                       for (j in (last_noNA_indices[i]+1):n)
-                               dat[j,i] = dat[last_noNA_indices[i],i]
-               }
+                       dat[(last_noNA_indices[i]+1):n,i] = dat[last_noNA_indices[i],i]
        }
 
        dat_with_date[,2:(p+1)] = dat