创建一个函数，该函数接受一个向量并在 R 中返回一个矩阵

Question

我正在尝试创建一个函数，该函数将接收向量 k 并返回给我一个长度为 length(distMat[1,]) 的矩阵，长度为 (k)。 distMat 是一个巨大的矩阵，而 indSpam 是一个长向量。特别是对于我的情况，长度（distMat [1，]）是2412。当我输入k作为长度为1的向量时，我得到一个长度为2412的向量。我希望能够输入k作为向量长度为 2，得到一个 2412x2 的矩阵。我正在尝试使用while循环让它通过k的长度，但它只返回给我一个长度为2412的向量。我做错了什么？

predNeighbor = function(k, distMat, indSpam){
  counter = 1
  while (counter<(length(k)+1))
  {  
preMatrix = apply(distMat, 1, order)
orderedMatrix = t(preMatrix)
truncate = orderedMatrix[,1:k[counter]]
checking = indSpam[truncate]
checking2 = matrix(checking, ncol = k[counter])
number = apply(checking2, 1, sum)


return(number[1:length(distMat[1,])] > (k[counter]/2))
counter = counter + 1
  }
}

Answer 1

我正在尝试创建一个函数，该函数将接受向量 k 并返回给我一个矩阵，其维度为 length(distMat[1,]) by length(k)

这是一个执行此操作的函数。

foo <- function(k, distMat) {
  return(matrix(0, nrow = length(distMat[1, ]), ncol = length(k)))
}

如果您有其他要求，请用文字描述。

根据您的评论，我想我更了解您的目标。您有一个返回长度为 k 的向量的函数，并且您希望将其输出保存为矩阵中的行。这是一个很常见的任务。让我们做一个简单的例子，k 以1:10 开头，然后说我们想用函数foo() 给它添加一些噪音，看看排名如何变化。

在函数的输入总是相同的情况下，replicate() 工作得很好。它会自动将所有内容放入矩阵中

k <- 1:10
noise_and_rank <- function(k) {
    rank(k + runif(length(k), min = -2, max = 2))
}

results <- replicate(n = 8, expr = {noise_and_rank(k)})

如果您想迭代，即一次的输出是下一次的输入，for 循环很好，我们只需预先分配一个矩阵用0，一次填写一列/行

k <- 1:10
n.sim <- 8
results <- matrix(0, nrow = length(k), ncol = n.sim)
results[, 1] <- k
for(i in 2:n.sim) {
    results[, i] <- noise_and_rank(results[, i - 1])
}

您最初的问题似乎是如何进行预分配。如果输入始终相同，则使用replicate() 意味着您不必担心。如果每次输入都不一样，则使用matrix()进行预分配，不需要编写任何特殊函数。