getsquares function

Generate Matrix of Squared Terms

Generate Matrix of Squared Terms

Takes a matrix of covariates and generates a new matrix that contains the original covariates and all squared terms. Squared terms for binary covariates are omitted.

getsquares(mat)

Arguments

  • mat: n by k numeric matrix of covariates.

Returns

n by k*2 numeric matrix that contains the original covariates plus all squared terms.

Author(s)

Jens Hainmueller

See Also

See matrixmaker

Examples

# create toy matrix mold <- replicate(3,rnorm(50)) colnames(mold) <- paste("x",1:3,sep="") head(mold) # create new matrix mnew <- getsquares(mold) head(mnew)