rpart情节文字较短

时间:2014-03-24 19:25:12

标签: r rpart

我正在使用prp包中的rpart.plot函数来绘制树。对于像状态这样的分类数据,它给出了很长的变量列表并使其可读性降低。如果超过一定长度,有没有办法将文本换行到两行或多行?

1 个答案:

答案 0 :(得分:3)

这是一个将长分割标签包裹在多个上的示例 线。每行的最大长度为25个字符。改变 25,以满足您的目的。 (这个例子来源于6.1节 rpart.plot vignette。)

tree <- rpart(Price/1000 ~ Mileage + Type + Country, cu.summary)

split.fun <- function(x, labs, digits, varlen, faclen)
{
    # replace commas with spaces (needed for strwrap)
    labs <- gsub(",", " ", labs)
    for(i in 1:length(labs)) {
        # split labs[i] into multiple lines
        labs[i] <- paste(strwrap(labs[i], width=25), collapse="\n")
    }
    labs
}

prp(tree, split.fun=split.fun)