“x”和“y”长度在自定义熵函数中不同

Question

我正在努力学习R，但我对它的工作方式有问题。我试图从头开始创建变量p和1-p的熵函数，但当我试图添加一些ifs以避免用0除法时出现nan时，我遇到了问题。

当我尝试使用plot的自定义熵时，它只是工作，但当我打印结果时，它会显示nan。但是当我尝试添加if时，它会说：

entropy <- function(p){

    cat("p = " , p)

    if (p==0 || p==1) {

       result = 0

    }else{

        result = - p*log2(p)-(1-p)*log2((1-p))

    }

    cat("\nresult=",result)

    return(result) 


}

p <- seq(0,1,0.01)

plot(p, entropy(p), type='l', main='Funcion entropia con dos valores posibles')

控制台不带ifs：

p=0 0.01 0.02 0.03 0.04 0.05 0.06 0.07 0.08 0.09 0.1 0.11 0.12 0.13 0.15 0.16 0.17 0.18 0.19 0.20.21 0.22 0.28 0.29 0.30.31 0.32 0.33 0.36 0.36 0.38 0.39 0.43 0.42 0.42 0.46 0.52 0.52 0.52 0.52 0.52 0.590.98 0.99 1

结果=NaN 0.08079314 0.1414405 0.1943919 0.2422922 0.286397 0.3274449 0.3659237 0.4021792 0.4364698 0.4364389 0.46842388 0.46843088 0.6577048 0.680077 0.7219281 0.741482727 0.7601675 0.7212786464646 0.882124646 0.882124646 0.886124508 0.886124508 0.886124508 0.886124509 0.886124509 0.8912909 0.8931735 0.904 3815 0.9149264 0.9248187 0.9340681 0.9426832 0.9506721 0.958042 0.96479506 0.9709506 0.986505 0.9827745 0.9974016 0.99827455 0.9974016 0.99827745 0.99927745 0.99927545 0.9895815 0.9895815 0.987505 0.99709506 0.9647995 0.958042 0.9506506 0.97095042 0.9506502 9426832 0.9340681 0.9248187 0.9248187 0.9149264 0.9043815 0.8931735 0.8812909 0.86872508 0.882674408 0.88127446 0.82127403 0.7219281 0.7219281 0.7219281 0.7219288 0.6573096 0.65742380.55842380.55842380.55842380.55842380.499990.4364690.4364690.4021790.3659237 0.3274449 0.286399 0.63574386 0.65742380.6574386 0.6574386 0.6574386 0.2422922 0.1943919 0.14144050.08079314NaN

使用if的控制台：

p=0 0.01 0.02 0.03 0.04 0.05 0.06 0.07 0.08 0.09 0.1 0.11 0.12 0.13 0.15 0.16 0.17 0.18 0.19 0.20.21 0.22 0.28 0.29 0.30.31 0.32 0.33 0.36 0.36 0.38 0.39 0.43 0.42 0.42 0.46 0.52 0.52 0.52 0.52 0.52 0.590.98 0.99 1 xy.coords（x,y,xlabel,ylabel,log）中的result=0error：“x”和“y”长度不同

Answer 1

您没有创建向量，而是创建了标量，因为您在if else子句中没有使用向量化功能。函数的结果只有一个数字。这应该有效：

entropy <- function(p){

  # initialize a vector of the desired length with zeros
  result <- numeric(length(p))

  # subset the vector for which you want to apply your formula on
  x <- p[!(p %in% c(0,1))]

  # overwrite only those positions for which you want to calculate values based
  # on your formula
  result[!(p %in% c(0,1))] <- - x*log2(x)-(1-x)*log2((1-x))


  #cat("\nresult=",result)

  return(result) 


}

p <- seq(0,1,0.01)

plot(p, entropy(p), type='l', main='Funcion entropia con dos valores posibles')