如何编写计算n步状态转移概率的R函数？

核心思路 #

要实现n步状态转移概率计算，关键是从状态序列中提取所有连续n个前置状态 + 1个后续状态的组合，统计每个前置状态组合对应各后续状态的出现次数，再计算占比得到转移概率。

完整函数代码 #

calc_n_step_trans_prob <- function(state_seq, n) {
  # 输入合法性校验
  if (!is.vector(state_seq) || length(state_seq) <= n) {
    stop("状态序列长度必须大于指定的n值")
  }
  
  # 生成所有长度为n+1的连续状态子序列
  total_groups <- length(state_seq) - n
  seq_groups <- lapply(1:total_groups, function(i) state_seq[i:(i+n)])
  
  # 拆分前置状态组合与后续状态
  trans_df <- do.call(rbind, lapply(seq_groups, function(group) {
    data.frame(
      prev_states = paste(group[1:n], collapse = "-"),
      next_state = group[n+1],
      stringsAsFactors = FALSE
    )
  }))
  
  # 统计转移次数与概率
  count_table <- table(trans_df$prev_states, trans_df$next_state)
  prob_table <- prop.table(count_table, margin = 1)  # 按行计算占比
  
  # 转换为易读的数据框格式
  library(reshape2)
  result <- melt(prob_table, varnames = c("前置状态组合", "后续状态"), value.name = "转移概率")
  result$转移次数 <- melt(count_table)$value
  
  # 按前置状态组合和后续状态排序
  return(result[order(result$前置状态组合, result$后续状态), ])
}

5步转移概率示例 #

# 模拟状态序列（状态取值：A/B/C）
set.seed(123)
state_seq <- sample(c("A", "B", "C"), size = 100, replace = TRUE)

# 计算5步转移概率
five_step_result <- calc_n_step_trans_prob(state_seq, n = 5)

# 查看前6条结果
head(five_step_result)

输出格式示例

内容的提问来源于stack exchange，提问作者heartofdarkness