Question

我想创建一个带有for循环的用户定义函数，该函数为每个列名称添加列号，以便为我的数据框添加清晰度。

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

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

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

Answer 1

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

但是，没有循环的更简单版本是使用setNames：

的矢量化方法

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

Answer 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

用户定义的函数和for循环

2 个答案: