我在 R 数据编程中遇到了 FOR 和 IF 函数的任务

Question

所以目前我遇到了 FOR 和 IF 函数的问题。

我有一个完整的工作代码，如下所示：

#my data input
demand.variation=c(1,4,20,23,37)
demand.probability=c(0.81,0.85,0.89,0.93,0.97)
n.month=25

#generating result matrix
result=matrix(0,n.month,2)
colnames(result)=c("Month","Demand")

#filling result matrix with month index and random demand value
result[,1]=1:n.month
random=runif(n.month,0,1)

#return demand value based on random value
for (i in seq_along(random)) {
  if (random[i]>demand.probability[5]) result[i,2]=demand.variation[5]
  else if (random[i]>demand.probability[4]) result[i,2]=demand.variation[4]
  else if (random[i]>demand.probability[3]) result[i,2]=demand.variation[3]
  else if (random[i]>demand.probability[2]) result[i,2]=demand.variation[2]
  else if (random[i]>demand.probability[1]) result[i,2]=demand.variation[1]
  else result[i,2]=0
}

#final result
result

结果完全可以使用：

> result
      Month Demand
 [1,]     1      0
 [2,]     2      0
 [3,]     3      0
 [4,]     4      0
 [5,]     5      0
 [6,]     6      0
 [7,]     7      0
 [8,]     8      0
 [9,]     9      4
[10,]    10      0
[11,]    11      0
[12,]    12     23
[13,]    13     37
[14,]    14      0
[15,]    15      1
[16,]    16      0
[17,]    17      0
[18,]    18      0
[19,]    19      1
[20,]    20     20
[21,]    21      0
[22,]    22      0
[23,]    23      0
[24,]    24      0
[25,]    25      0

问题是，我必须重新运行相同的代码，但 demand.variation 和 demand.probability 的长度和值不同，所以我将面临这两个变量的几十个变化的问题，因为 FOR 和 IF 是手动输入。

我的问题是如何生成灵活的 FOR 和 IF 语句代码来覆盖#return demand value based on random value 代码的工作。

脚注：demand.variation 和 demand.probability 将始终具有相同的向量长度，并相应地相互连接，这意味着 demand.variation 向量中的第一个值有demand.probability 向量中的第一个值的概率，等等

demand.variation=c(1,4,20,23,37)
demand.probability=c(0.81,0.85,0.89,0.93,0.97)

如果您能在这方面帮助我，我将不胜感激。

Answer 1

您可以添加另一个 for() 循环，从 demand.variation 向量中的条目数倒数。

for(i in seq_along(random)){
  for(j in seq_along(demand.variation):1){
    if(random[i] > demand.probability[j]){
      result[i,2] <-  demand.variation[j]
    } else {
      result[i,2] <- 0
    }
  }
}

编辑：澄清：

所以我把你想要的东西分解成几个函数，并组合成一个主函数，可以根据需要的任何长度/组合调用：

create.demand.variation <- function(vector){
  return(demand.probability <- c(vector))
}

create.demand.probability <- function(vector){
  return(demand.probability <- c(vector))
}

create.random <- function(n.month){
  return(random <- runif(n.month, 0, 1))
}

create.result <- function(n.month){
  result=matrix(0,n.month,2)
  colnames(result)=c("Month","Demand")
  result[,1]=1:n.month
  return(result)
}

create.random <- function(n.month){
  return(random <- runif(n.month,0,1))
}

get.result <- function(demand.variation, demand.probability, random.vector, result){
  for(i in 1:length(random)){
    k <- length(demand.probability)
    while(k >= 1){
      if(random[i] > demand.probability[k]){
        result[i,2] <- demand.variation[k]
        k <- 0
      } else {
        k <- k - 1
      }
    }
  }
  return(result)
}

master.function <- function(vector.demand, vector.prob, n.month ){
  create.demand.variation(vector.demand) -> demand.variation
  create.demand.probability(vector.prob) -> demand.probability
  create.random(n.month) -> random.vector
  create.result(n.month) -> result
  get.result(demand.variation, demand.probability, random.vector, result) -> result
  return(result)
}

所以对于示例向量，您可以输入：

master.function(c(1,4,20,23,37), c(0.81, 0.85, 0.89, 0.93, 0.97), 25) -> 
result

对我来说，这会产生：

> result
      Month Demand
 [1,]     1      0
 [2,]     2     20
 [3,]     3      0
 [4,]     4      0
 [5,]     5      0
 [6,]     6      0
 [7,]     7      1
 [8,]     8      0
 [9,]     9      4
[10,]    10      0
[11,]    11     20
[12,]    12      0
[13,]    13     23
[14,]    14      0
[15,]    15      0
[16,]    16      0
[17,]    17      0
[18,]    18      0
[19,]    19      0
[20,]    20      0
[21,]    21      0
[22,]    22      0
[23,]    23      0
[24,]    24      0
[25,]    25      1

Answer 2

您正在尝试按区间对连续值（在 random 中）进行分类，区间由 demand.probability 给出，并且应该标记为 demand.variation 中给出的值。 这可以通过 R 中的 cut() 函数来完成。 但是，您必须添加一些额外的边框（-Inf、Inf 和例如值 0）。

只需将您的 for-loop 大表达式替换为：

result[, 2] <- cut(random, breaks=c(-Inf, demand.probability, Inf), labels=c(0, demand.variation))

您可以将其作为一个函数来执行（带有可重复性的种子）：

categorize_months <- function(n_months, breaks, labels, seed=42) {
  set.seed(seed)
  random <- runif(n_months, 0, 1)
  months <- 1:n_months
  demands <- cut(random, breaks=c(-Inf, breaks, Inf), labels=c(0, labels))
  df <- data.frame("Month"=months, "Demand"=demands)
  df
}

categorize_months(25, demand.probability, demand.variation, seed=42)