如何调节函数的一部分

Question

我有这个函数稍后将应用于 3D 数组f3 <- function(x) {-sum((x / sum(x)) * log2(x / sum(x)))}，我正在尝试应用条件，如果函数的第二部分是-Inf（即log2(x/sum(x)=-Inf），那么这部分将是0（即log2(x/sum(x))=0)。我正在尝试在函数内嵌套一个 if 语句（见下文）但暂时不起作用。

f3 <- function(x) {
-sum((x / sum(x)) * log2(x / sum(x)))
  if((log2(x/sum(x)))==-Inf) {
log2(x/sum(x))==0
}
}

我在 1000 个 24x24 矩阵 (a) 的 3D 数组上使用这个函数来尝试这个我，我只使用数组 a[,,1] 的第一个矩阵

apply (a[,,1],2,f3)

欢迎任何小提示。 提前致谢。

Answer 1

我相信您正在寻找的东西可以像这样完成：

f3 <- function(x) {
  if(log2(x/sum(x)) == -Inf) {
    result <- 0
  } else {
    result <- -sum((x / sum(x)) * log2(x / sum(x)))
  }
  return(result)
}

Answer 2

也许这对你有用。只是微小的变化：

f3 <- function(x) {
  denom <- log2(x / sum(x))
  if(denom==-Inf|is.nan(denom)) { denom <- 0}
  -sum((x / sum(x)) * denom)
}

注意：请记住，=（单等号）是等效于 <- 的赋值运算符。对于逻辑比较，您应该使用==（双等号）