r - 在多个数据框中创建新列，其值取决于数据框的名称

Question

我在 R 中有几个类似命名的数据框。我正在寻找一种巧妙的方法来在每个数据框中创建一个新列。这些列中的值应取决于数据框的名称。

我有：

dfARCHIMEDES
    var1   var2
     a      b
     c      d

dfPYTHAGORAS
    var1   var2
     e      f
     g      h

它应该变成：

dfARCHIMEDES
    var1   var2   newvar
     a      b      ARCHIMEDES
     c      d      ARCHIMEDES

dfPYTHAGORAS
    var1   var2   newvar
     e      f      PYTHAGORAS
     g      h      PYTHAGORAS

不用说，如果数据如此简单，那将是一件容易的事。但是，我目前有大约 250 个数据帧，每个数据帧有 15 列和 500 行。

如果有任何帮助，我将不胜感激。

Answer 1

您可以通过以下方式解决此问题。请参阅代码中的注释。

# make some sample data (which OP should provide...)
dfARCHIMEDES <- data.frame(var1 = c("a", "c"),
                           var2 = c("b", "d"))
dfPYTHAGORAS <- data.frame(var1 = c("e", "g"),
                           var2 = c("g", "h"))

# collect all object names that start with a "df" from your workspace
get.dfs <- ls(pattern ="^df")

# go through each data.frame
for (i in get.dfs) {
  tmp <- get(i) # save the data.frame into a temporary variable
  newname <- gsub("^df", "", i) # remove the df part from the name
  tmp$newvar <- newname # create new variable with the new name
  assign(i, tmp) # re-write the data.frame
}

> dfARCHIMEDES
  var1 var2     newvar
1    a    b ARCHIMEDES
2    c    d ARCHIMEDES
> dfPYTHAGORAS
  var1 var2     newvar
1    e    g PYTHAGORAS
2    g    h PYTHAGORAS

Answer 2

您可以通过执行以下操作获取具有全局环境中所有数据框名称的向量：

library(dplyr)

df_names <- 
  sapply(.GlobalEnv, is.data.frame) %>% 
  which() %>% 
  names()

之后，一个循环就可以解决问题：

for (i in df_names) {
  assign(i, cbind(get(i), newvar = i))
}