如何获取partykit中每个节点的路径/规则（含非终端节点）

你可以通过遍历决策树的所有节点，递归或循环生成每个节点从根节点到该节点的路径规则。以下是两种实用方法，适配你用rpart转成partykit对象的场景：

方法1：基于rpart的路径提取（适合rpart转来的party对象） #

利用rpart的path.rpart()函数获取每个节点的路径，再拼接成规则字符串：

library(partykit)
library(rpart)
library(MASS)

# 补全示例代码的参数（原代码缺少n和p的定义）
n <- 1000
p <- 5

X    <- MASS::mvrnorm(n, rep(0, p), diag(p))
y    <- as.numeric(drop(X %*% rep(1, p)) > 2)
data <- data.frame(y, X)

tree  <- rpart(y ~ ., data = data, control = rpart.control(cp = 0.005))
pfit  <- as.party(tree)

# 自定义函数：获取所有节点的规则
get_all_node_rules <- function(party_tree) {
  all_nodes <- nodeids(party_tree)
  node_rules <- list()
  
  for (node in all_nodes) {
    # 获取当前节点的路径（从根到该节点的所有条件）
    path <- path.rpart(as.rpart(party_tree), node)
    # 拼接成规则字符串
    rule <- paste(names(path[[1]]), path[[1]], collapse = " & ")
    # 根节点无规则，单独标注
    if (rule == "") rule <- "Root node (no conditions)"
    node_rules[[as.character(node)]] <- rule
  }
  return(node_rules)
}

# 调用函数获取结果
all_node_rules <- get_all_node_rules(pfit)

# 查看所有节点规则
all_node_rules

方法2：纯partykit原生方法（通用所有party对象） #

直接遍历partykit的节点结构，递归生成路径规则，支持数值变量和分类变量的分割：

# 自定义函数：纯partykit实现的规则提取
get_all_node_rules_party <- function(party_tree) {
  # 递归获取单个节点的路径规则
  get_node_path <- function(node) {
    if (node$id == 1) {
      return("Root node (no conditions)")
    }
    # 获取父节点
    parent_node <- party_tree[node$parent]
    split_info <- parent_node$split
    var_name <- names(split_info$variable)
    
    # 根据分割类型生成条件
    if (split_info$type == "numeric") {
      split_val <- split_info$breaks
      # 判断当前节点属于父节点的左分支还是右分支
      cond <- if (node$id %in% parent_node$kids[[1]]$id) {
        paste(var_name, "<=", round(split_val, 4))
      } else {
        paste(var_name, ">", round(split_val, 4))
      }
    } else {
      # 分类变量的分割条件
      target_levels <- split_info$levels[node$id %in% parent_node$kids[[1]]$id]
      cond <- paste(var_name, "%in% c('", paste(target_levels, collapse = "', '"), "')")
    }
    # 递归拼接父节点的规则
    paste(get_node_path(parent_node), " & ", cond)
  }
  
  # 遍历所有节点
  all_nodes <- nodeids(party_tree)
  node_rules <- lapply(all_nodes, function(id) {
    get_node_path(party_tree[id])
  })
  names(node_rules) <- all_nodes
  return(node_rules)
}

# 调用函数
all_node_rules_party <- get_all_node_rules_party(pfit)

# 查看结果
all_node_rules_party

两种方法都会返回一个列表，键是节点ID，值是对应节点的完整路径规则，包含所有非终端节点的条件。

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