web-dev-qa-db-ja.com

変数のベクトルをlm()式に渡します

プログラミングが面倒にならないように、コードの一部を自動化しようとしていました。

基本的に、rmsパッケージのfastbw()を使用して、変数の段階的な選択を試みていました。 fastbw()で選択された変数のリストをy ~ x1+x2+x3として式に渡したいと思います。 "x1" "x2" "x3"はfastbw()で選択された変数のリストです

ここに私が試したがうまくいかなかったコードがあります

olsOAW0.r060 <- ols(roll_pct~byoy+trans_YoY+change18m, 
                    subset= helper=="POPNOAW0_r060", 
                    na.action = na.exclude, 
                    data = modelready)

OAW0 <- fastbw(olsOAW0.r060, rule="p", type="residual", sls= 0.05)

vec <- as.vector(OAW0$names.kept, mode="any")

b <- paste(vec, sep ="+") ##I even tried b <- paste(OAW0$names.kept, sep="+")

bestp.OAW0.r060 <- lm(roll_pct ~ b , 
                      data = modelready, 
                      subset = helper =="POPNOAW0_r060",    
                      na.action = na.exclude)

私はRを初めて使用しますが、まだ急な学習曲線をたどっていません。そのため、明らかなプログラミングの失敗をおizeびします。

41
Anand

あなたはほとんどそこにいます。単にpaste全体式をまとめて、次のようにする必要があります。

paste("roll_pct ~ ",b,sep = "")

as.formulaを使用して実際の式に強制してから、thatlmに渡します。技術的には、lmは文字列自体を強制するかもしれませんが、自分で強制する方が一般に安全です。 (式を期待する関数の中には、強制を行わないものもあれば、そうでないものもあります。)

58
joran

Bを定義するとき、実際にはsebの代わりにcollapseを使用する必要があります。

b <- paste(OAW0$names.kept, collapse="+")

その後、ジョランの答えにそれを置くことができます

paste("roll_pct ~ ",b,sep = "")

または単に使用する:

paste("roll_pct ~ ",paste(OAW0$names.kept, collapse="+"),sep = "")
19
cconnell

今日、同様の問題に遭遇しました。クラス名を固定する必要さえない、より一般的なものにしたい場合は、

frmla <- as.formula(paste(colnames(modelready)[1], paste(colnames(modelready)[2:ncol(modelready)], sep = "", 
                              collapse = " + "), sep = " ~ "))

これは、最初の列にクラス変数または従属変数があることを前提としていますが、次のようにインデックス付けを最後の列に簡単に切り替えることができます。

frmla <- as.formula(paste(colnames(modelready)[ncol(modelready)], paste(colnames(modelready)[1:(ncol(modelready)-1)], sep = "", 
                              collapse = " + "), sep = " ~ "))

次に、次を使用してlmに進みます。

bestp.OAW0.r060 <- lm(frmla , data = modelready, ... )
8
discipulus

冗長性の少ないものを探している場合:

fm <- as.formula( paste( colnames(df)[i], ".", sep=" ~ ")) 
                                      # i is the index of the outcome column

これは関数内にあります:

getFormula<-function(target, df) {

  i <- grep(target,colnames(df))
  as.formula(paste(colnames(df)[i], 
                   ".", 
                   sep = " ~ "))
}
fm <- getFormula("myOutcomeColumnName", myDataFrame)
rp <- rpart(fm, data = myDataFrame) # Use the formula to build a model
1
Travis Heeter

関数に基づいて、上記の答えを単純化して収集するためだけに

my_formula<- function(colPosition, trainSet){
    dep_part<- paste(colnames(trainSet)[colPosition],"~",sep=" ")
    ind_part<- paste(colnames(trainSet)[-colPosition],collapse=" + ")
    dt_formula<- as.formula(paste(dep_part,ind_part,sep=" "))
    return(dt_formula)
}

使用するには:

my_formula( dependent_var_position, myTrainSet)
0
user3169330