[R] simplify code for dummy coding of factors
Richard M. Heiberger
rmh at temple.edu
Wed Dec 31 04:16:27 CET 2014
It can handle one more simplification step
# Rich Heiberger, removing intercept, including haireye data
haireye.df <- cbind(
as.data.frame(haireye),
model.matrix(Freq ~ Hair + Eye, data=haireye,
contrasts.arg=list(Hair=diag(4), Eye=diag(4)))[,-1]
)
haireye.df
On Tue, Dec 30, 2014 at 9:15 PM, Michael Friendly <friendly at yorku.ca> wrote:
> On 12/30/14 7:16 PM, Richard M. Heiberger wrote:
>>
>> I like this very simple version. Note that you don't need
>> as.data.frame().
>>
>> model.matrix(Freq ~ Hair + Eye, data=haireye,
>> contrasts.arg=list(Hair=diag(4), Eye=diag(4)))
>
> Thanks to all who replied to this question. model.matrix() was what I had
> missed, but to
> get the result in the form I wanted, required a bit more work. These
> solutions give me
> what I asked for.
>
> haireye <- margin.table(HairEyeColor, 1:2)
>
> # Jeff Newmiller, solution 2, corrected
>
> haireye.df <- as.data.frame(haireye)
> dummykeys <- data.frame( h = factor( as.integer( haireye.df$Hair ) )
> , e = factor( as.integer( haireye.df$Eye ) ) )
> dummy.hair <- as.data.frame( model.matrix( ~ h - 1, data=dummykeys ))
> dummy.eye <- as.data.frame( model.matrix( ~ e - 1, data=dummykeys ))
> haireye.df <- data.frame( haireye.df, dummy.hair, dummy.eye )
>
>
> # Rich Heiberger, removing intercept, including haireye data
> haireye.df <- as.data.frame(haireye)
> haireye.df <- cbind(
> haireye.df,
> model.matrix(Freq ~ Hair + Eye, data=haireye,
> contrasts.arg=list(Hair=diag(4), Eye=diag(4)))[,-1]
> )
> haireye.df
>
>
>
>
> --
> Michael Friendly Email: friendly AT yorku DOT ca
> Professor, Psychology Dept. & Chair, Quantitative Methods
> York University Voice: 416 736-2100 x66249 Fax: 416 736-5814
> 4700 Keele Street Web: http://www.datavis.ca
> Toronto, ONT M3J 1P3 CANADA
>
More information about the R-help
mailing list