R lapply 具有两个不固定参数的函数

R lapply function with two arguments that are not fixed

我找到一个之前问过的。我的问题比上一个问题复杂一点。对于我的问题,y参数不固定。

在函数(X,Y){SOME FUNCTION}中,X是字符列表,Y是数据帧列表。基本上,我希望函数分别按顺序处理 X 和 Y 对,并将输出生成为一个列表。比如X列表的第一个元素和Y列表的第一个元素,X列表的第二个元素和Y列表的第二个元素,X列表的第三个元素和Y列表的第三个元素,...

X、Y 的例子

X <- c("1", "2")
y1 <- data.frame("person.1" = "Amy", "bestfood..1" = "fish", "bestthing..1" = "book",
                 "person.2" = "Mike", "bestfood..2" = "fish", "bestthing..2" = "book")
y2 <- data.frame("person.1" = "Amy","bestfood..1" = "carrot", "bestthing..1" = "cloth",
                 "person.2" = "Mike","bestfood..2" = "carrot", "bestthing..2" = "cloth")
Y <- list(y1,y2)

函数:

  addID <- function(X, Y) {
     rowlength <- length(Y)
     df <- as.data.frame(matrix(NA, nrow = rowlength, ncol = 3))
     colnames(df) <- c("ID", "Person", "Food")
     df[1:nrow(df), 1] <- X

  # name
  namecols <-grep("person",colnames(Y))
  for (i in 1:length(namecols)) {
    name <- Y[1, namecols[i]]
    df[i, 2] <- as.character(name)
  }
  # food
  foodcols <-
  grep("bestfood",colnames(Y))
  for (i in 1:length(foodcols)) {
    food <- Y[1, foodcols[i]]
    df[i, 3] <- as.character(foodcols)
  }

  return(df)
   }
  }

我尝试使用 lapply 但无法找到包含 X 列表的方法。当我尝试这个时:

lapply(Y, function, X=X)

该功能无法正常使用。我想知道是否有其他方法可以在其中包含 X(我在单个字符和数据帧上尝试了该功能,它工作得很好。)

我希望这是清楚的。如果不是,请指出,我会尽力澄清。提前致谢。


更新:

我按照评论的建议尝试了地图。它 returns: 维数不正确。我在函数中添加了一些细节。好像 R 卡在了最后一行。

outcome <- Map(addID, Y, X)

我明白了

error in Y[1, namecols[i]] : incorrect number of dimensions
In addition: Warning message:
In `[<-.data.frame`(`*tmp*`, 1:nrow(df), 1, value = list(person.1 = 1L,  :
 provided 6 variables to replace 1 variables

结果应如下所示:

z1 <- data.frame(ID = c(1,2), Person = c("Amy","Mike"), Food = c("fish", "fish"))
z2 <- data.frame(ID = c(1,2), Person = c("Amy","Mike"), Food = c("carrot", "carrot"))
outcome <- list(z1,z2)

我们可以在 tidyverse

中轻松做到这一点
library(dplyr)
library(tidyr)
bind_rows(Y, .id = 'ID') %>% 
    select(ID, starts_with('person'),  contains('food')) %>% 
    pivot_longer(cols = -ID, names_to = c(".value"),
       names_pattern = "([^.]+)\.+\d+")

-输出

# A tibble: 4 x 3
  ID    person bestfood
  <chr> <chr>  <chr>   
1 1     Amy    fish    
2 1     Mike   fish    
3 2     Amy    carrot  
4 2     Mike   carrot  

有了OP的功能,我们修改一下就可以了

addID <- function(X, Y) {
     rowlength <- length(Y)
     df <- as.data.frame(matrix(NA, nrow = rowlength, ncol = 3))
     colnames(df) <- c("ID", "Person", "Food")
     df[1:nrow(df), 1] <- X
     namecols  <- grep("person",colnames(Y))
     df[, 2] <- unlist(Y[namecols])
     foodcols <- grep("bestfood", colnames(Y))
     df[,3] <- unlist(Y[foodcols])
     

  

  return(unique(df))
   }

-测试

Map(addID, X, Y)
$`1`
  ID Person Food
1  1    Amy fish
2  1   Mike fish

$`2`
  ID Person   Food
1  2    Amy carrot
2  2   Mike carrot