Сохранение терминального узла правила разделения ctree в dataframe в R

Question

У меня есть набор данных (Node1), который содержит 6 брендов и применяет ctree(), то же самое разделяется на 2 терминальных узла. Node2 содержит 4 торговых марок Node3 содержит 2 торговых марок. Я хотел бы извлечь эти бренды каждого терминального узла и сохранить их в двух разных фреймах. Пожалуйста, предложите, как сделать то же самое.

Это то, что я закодированы:

library("party") 
    library(gridExtra) 
    fileName <- "C\\" 
    data <- read.csv(paste(fileName, ".csv", sep=""), header=TRUE) 
    pdd <- subset(data, select=c(col1,col2)) 

    pdd_ctree <- ctree(col1~col2, data=pdd, controls = ctree_control(minsplit=30)) 

    print(pdd_ctree@tree$psplit$splitpoint[1:6]) 
    print(pdd_ctree@tree$psplit$splitpoint) 

В результате я получаю:

print(pdd_ctree@tree$psplit$splitpoint[1:6]) 
    [1] 1 0 1 1 0 1 

    print(pdd_ctree@tree$psplit$splitpoint) 
    [1] 1 0 1 1 0 1 
    attr(,"levels") 
    [1] "Brand01"   "Brand02"    "Brand03"   "Brand04"    
    [5] "Brand05" "Brand06" 

Мое требование:

Есть 2 dataframes left.df и right.df

левый.df будет содержать [Brand01, Brand03, Brand04, Brand06]

right.df будет содержать [Brand02, Brand05]

Answer 1

Я думаю, что вы просто хотите:

lvls <- levels(pdd_ctree@tree$psplit$splitpoint) 
left.df = lvls[pdd_ctree@tree$psplit$splitpoint == 1] 
right.df = lvls[pdd_ctree@tree$psplit$splitpoint == 0]