2017-06-07 10 views
1

pythonでは、回帰にカテゴリ値を使用してカテゴリ値に対して0と1の行列を作成するときに、patsyモジュールのdmatrices関数を使用できます。R:カテゴリの値のためにRにdmatrices(pythonから)のようなものがありますか?

Rには同じ機能を果たすライブラリ/関数がありますか?

+1

'?model.matrix'。これは重複している可能性があります... –

+0

https://stackoverflow.com/questions/5048638/automatically-expanding-an-r-factor-into-a-collection-of-1-0-indicator-variables/5048726#5048726 –

+3

また、ほとんどの回帰Rのモデリング関数は実際にはこれを行う必要はありません。なぜなら、因子は 'contrasts'属性に基づいて自動的に処理されるからです。 – Marius

答えて

2

このデータフレームでは、列xとyが数値であり、列fが係数であるとします。その後、我々はこのように回帰を実行することができますし、lmは0/1カラムを含む適切なモデル行列に式を変換し、その上で回帰を実行します:

# test data 
set.seed(123) 
DF <- transform(data.frame(f = gl(3, 5, labels = letters[1:3]), x = 1:15), 
     y = rnorm(15, 1:15)) 

# run regression 
fo <- y ~ x + f 
lm(fo, DF) 

モデル行列がそのようにそこに上記を行う際に計算され、明示的に計算する必要はありませんが、これを試したい場合は、これを試してみてください:

# view model matrix 
model.matrix(fo, DF) 
+0

それは素晴らしいです!ありがとうございました - 正確に私が必要なもの – jeangelj

+0

ああ、統計用に書かれた言語を使用する魔法... – Alex

関連する問題