r - apply() 函数分别按顺序应用于矩阵的列和数值向量的元素

Question

我有一个函数 f(list,t)，其中第一个参数是 list，第二个参数 t 是一个数字。我想分别将 f 应用于矩阵 M 的列和向量 T 的元素。因此，如果 M 的列是 (M_1,M_2,...,M_k) 和 T = (t_1,t_2,...,t_k)，我想得到以下内容:

f(M_1,t_1), f(M_2,t_2), ..., f(M_k,t_k)。

有没有不使用 for 循环的有效方法？

例如如果

f <- function(list,x) {x %in% list}
M <- matrix(1:12,4,3)
T <- c(1,2,10)

我希望得到

TRUE FALSE TRUE

以下行将 f 应用于 M 的每一列和 T 的每个元素

apply(M,2,f,T)

但是我需要的只是这个输出的对角线，所以我想要一种方法来避免额外的计算。

Answer 1

您还可以使用 sapply 使用矩阵中的列数。稍后，我们使用 any 从每一列返回 True(如果有)值

Tr <- c(1,2,10)
sapply(seq(ncol(M)), function(x) any(f(M[,x], Tr)))

[1]  TRUE FALSE  TRUE