如何在R中匹配矩阵计算后Omega向量对应起始状态的元素？

要适配任意规模的转移概率矩阵，核心是动态识别吸收态与非吸收态，并通过映射关系自动匹配起始状态对应的Omega向量元素，而非依赖固定位置偏移（如start-1）。以下是具体实现方案：

核心步骤说明 #

1. 自动识别吸收态 #

吸收态的判定规则是：转移矩阵对角线上元素为1（自身转移概率为1）。通过该规则可动态提取任意矩阵中的吸收态：

absorbing_states <- which(diag(P) == 1)
non_absorbing_states <- setdiff(1:nrow(P), absorbing_states)

2. 构建子矩阵Q与R #

• Q矩阵：非吸收态之间的转移概率子矩阵
• R矩阵：非吸收态到各吸收态的转移概率子矩阵

Q <- P[non_absorbing_states, non_absorbing_states]
R <- P[non_absorbing_states, absorbing_states]

3. 计算Omega矩阵 #

通过基本矩阵N = (I - Q)^-1计算Omega，Omega的每一行对应一个非吸收态，每一列对应一个吸收态，元素值为从该行非吸收态出发，首次到达该列吸收态的概率：

N <- solve(diag(nrow(Q)) - Q)
Omega <- N %*% R

4. 动态映射起始状态 #

使用match()函数定位起始状态在non_absorbing_states中的位置，无需手动指定偏移：

start_row <- match(start_state, non_absorbing_states)

完整通用函数 #

first_passage_prob <- function(P, start_state, target_absorb, competing_absorb) {
  # 识别吸收态与非吸收态
  absorbing_states <- which(diag(P) == 1)
  non_absorbing_states <- setdiff(1:nrow(P), absorbing_states)
  
  # 处理起始状态为吸收态的边界情况
  if (start_state == target_absorb) {
    return(1)
  } else if (start_state == competing_absorb) {
    return(0)
  } else if (!(start_state %in% non_absorbing_states)) {
    stop("起始状态不是非吸收态或指定的吸收态")
  }
  
  # 构建Q、R子矩阵
  Q <- P[non_absorbing_states, non_absorbing_states]
  R <- P[non_absorbing_states, absorbing_states]
  
  # 计算Omega矩阵
  N <- solve(diag(nrow(Q)) - Q)
  Omega <- N %*% R
  
  # 定位目标与竞争吸收态的列索引
  target_col <- match(target_absorb, absorbing_states)
  
  # 定位起始状态对应的行索引
  start_row <- match(start_state, non_absorbing_states)
  
  # 返回首次到达目标吸收态先于竞争态的概率
  return(Omega[start_row, target_col])
}

测试示例 #

用你提供的转移矩阵测试从状态3出发，首次到达状态5先于状态1的概率：

# 定义转移矩阵P
P <- matrix(c(1,0,0,0,0,
              0.4,0,0.6,0,0,
              0,0.4,0,0.6,0,
              0,0,0.4,0,0.6,
              0,0,0,0,1),5,5,byrow = TRUE)

# 调用函数
first_passage_prob(P, start_state = 3, target_absorb = 5, competing_absorb = 1)

内容的提问来源于stack exchange，提问作者Homer Jay Simpson