用户定义函数和for循环答案

【问题标题】：user-defined function and for loop用户定义函数和for循环
【发布时间】：2014-12-17 18:46:31
【问题描述】：

我想创建一个带有 for 循环的用户定义函数，该循环将在每个列名中添加列号，以使我的数据框更加清晰。

这是代码（但不起作用）：

addncol = function(df)
   {
    for(i in 1:ncol(df))
       {
        names(df)[[i]] <- paste(names(df)[[i]], i, sep = '_')
       }
   }

单独的 for 循环运行良好，但在函数中它不起作用（虽然我没有错误消息）。

【问题讨论】：

names(df) 不是一个列表，而是一个向量。以这种方式使用它会更典型：........names(df)[i] <- ...。但这不是根本错误，那么您还需要将其返回并将其分配回原始df，因为您刚刚所做的只是在函数内部发生。或者你可以使用<<-
谢谢斯文。有什么方法可以在函数内部自动将其分配回 df ？而不是在函数外输入：df <- addncol(df)，我只想输入：addncol(df) 并且我的新df被修改了，控制台中没有显示
不确定 Sven 是谁，但我已经提到了<<-。行家认为这是作弊。

标签： r for-loop user-defined-functions

【解决方案1】：

函数不返回数据框df。如果您将df 添加为函数的最后一行，它将起作用。

但是，没有循环的更简单版本是带有setNames 的矢量化方法：

addncol <- function(df)
   setNames(df, paste(names(df), seq(df), sep = "_"))

【讨论】：

setNames 确实是一个更好的选择。

【解决方案2】：

您可以为此使用 data.tables，它使用 setnames(...) 通过引用更新（注意不同的拼写 - 它是不同的函数）。

library(data.table)
DT <- as.data.table(mtcars)   # example only
head(DT)
mpg cyl disp  hp drat    wt  qsec vs am gear carb
# 1: 21.0   6  160 110 3.90 2.620 16.46  0  1    4    4
# 2: 21.0   6  160 110 3.90 2.875 17.02  0  1    4    4
# 3: 22.8   4  108  93 3.85 2.320 18.61  1  1    4    1
# 4: 21.4   6  258 110 3.08 3.215 19.44  1  0    3    1
# 5: 18.7   8  360 175 3.15 3.440 17.02  0  0    3    2
# 6: 18.1   6  225 105 2.76 3.460 20.22  1  0    3    1
addncol <- function(DT) setnames(DT,paste(names(DT),seq(DT),sep="_"))
addncol(DT)
head(DT)
#    mpg_1 cyl_2 disp_3 hp_4 drat_5  wt_6 qsec_7 vs_8 am_9 gear_10 carb_11
# 1:  21.0     6    160  110   3.90 2.620  16.46    0    1       4       4
# 2:  21.0     6    160  110   3.90 2.875  17.02    0    1       4       4
# 3:  22.8     4    108   93   3.85 2.320  18.61    1    1       4       1
# 4:  21.4     6    258  110   3.08 3.215  19.44    1    0       3       1
# 5:  18.7     8    360  175   3.15 3.440  17.02    0    0       3       2
# 6:  18.1     6    225  105   2.76 3.460  20.22    1    0       3       1

【讨论】：