R语言使用for循环生成行业生产预测矩阵报错求助

存在的问题 #

• 矩阵初始化逻辑错误：matrix(length(products):length(column.name)) 未指定行列参数，生成的矩阵结构完全不符合3行（产品数）10列（年份数）的预期
• 变量名拼写错误：内层循环调用了未定义的product变量，实际定义的产品向量为products，拼写不一致导致取值失败
• 错误使用return语句：return仅支持在R函数内部使用，写在全局脚本的循环中会导致代码执行完第一次外层循环就强制终止，无法完成所有列的赋值
• 语法冗余：定义products向量时末尾多了多余逗号，虽不触发报错但不符合规范

修复后完整代码 #

# 生成年份列名
column.name <- vector()
for (i in 1:10) {
    object <- paste('ano.',i)
    column.name[i] <- object
}

# 定义产品向量，删除冗余逗号
products <- c('a','b','c')

# 产能参数配置
participation <- c(0.60,0.20,0.20)
amount <- 4800

# 正确初始化3行10列的矩阵，提前设置行名列名
prod <- matrix(
  nrow = length(products), 
  ncol = length(column.name),
  dimnames = list(products, column.name)
)

# 嵌套循环完成赋值，修正变量名、删除非法return语句
for (j in 1:length(column.name)) {
  for (i in 1:length(products)) {
    dado <- participation[i] * amount
    prod[i,j] <- dado
  }
}

# 输出验证结果
prod

优化方案 #

该场景无需使用嵌套循环，利用R的向量化特性可一行代码完成矩阵生成，执行效率更高：

prod <- matrix(
  participation * amount, 
  nrow = length(products), 
  ncol = length(column.name),
  dimnames = list(products, column.name)
)

内容的提问来源于stack exchange，提问作者vini.m.oliveira