用列索引而不是名称定义rpart预测变量

Question

我有以下代码行：

my_dt1 <- rpart(Survived ~ Pclass + Sex + Age + SibSp + Parch + Fare + Embarked, 
                  data = train2, 
                  method = "class")

我想用Pclass中的相应列索引而不是列名来调用那些预测变量（Sex，Age，train2，...）使代码更具通用性。

我尝试将Pclass的{​​{1}}更改，但出现此错误：

train2[,3]

Answer 1

对示例数据有效：

user-agent

或：

df <- as.data.frame(cbind(rnorm(100, 2, 1),
rnorm(100,5,2),
rnorm(100,3,1)))

mod <- rpart(df[,1] ~ df[,2] + df[,3], 
       method = "class")

mod