时间戳

首页 > 解决方案 > 将函数应用于来自不同列表的元素时出现问题

r - 将函数应用于来自不同列表的元素时出现问题

问题描述

我想setdiff使用两个列表中的元素进行计算。

为此，我创建了一个在提供参数时可以正常工作的函数，但在我尝试使用mapply.

# The lists have the following structure
list_A <- list(
  vec_A1 = list(v1="1",v2="2",v3="3"),
  vec_A2 = list(v1="3",v2="4",v3="5")
  )

list_B <- list(
  mat_B1 = matrix(as.character(1:3),nrow = 3,ncol = 1),
  mat_B2 = matrix(as.character(3:5),nrow = 3,ncol = 1)  
)

myfun <- function(vec,mat){
  vec = unlist(vec) # Create a vector from the list's elements
  mat = mat[[1]] # Extract the required matrix
  x = apply(mat,1,base::setdiff,x=vec) %>% t # Compare the vector with each matrix row
  return(x)
}

# It gives my desired output when applied over single elements from the lists
myfun(list_A[1],list_B[1])
myfun(list_A[2],list_B[2])

# But fails when using mapply
mapply(myfun,list_A,list_B, SIMPLIFY = F)

所需的输出是

     [,1] [,2]
[1,] "2"  "3" 
[2,] "1"  "3" 
[3,] "1"  "2" 

     [,1] [,2]
[1,] "4"  "5" 
[2,] "3"  "5" 
[3,] "3"  "4"

但是有了 mapply 我得到了

 Error in apply(mat, 1, base::setdiff, x = vec) : 
  dim(X) must have a positive length

关于我缺少什么的任何提示？

提前致谢。

标签： rmapply

解决方案

我想你只是混淆了[和[[。试试这个：

myfun <- function(vec,mat){
    vec = unlist(vec) # Create a vector from the list's elements
    #mat = mat[[1]] # Extract the required matrix
    x = apply(mat,1,base::setdiff,x=vec) %>% t # Compare the vector with each matrix row
    return(x)
}

# It gives my desired output when applied over single elements from the lists
myfun(list_A[[1]],list_B[[1]])
myfun(list_A[[2]],list_B[[2]])

# But fails when using mapply
mapply(myfun,list_A,list_B, SIMPLIFY = F)

推荐阅读