mirror of
https://github.com/agdamsbo/daDoctoR.git
synced 2024-11-22 11:50:23 +01:00
90 lines
2.3 KiB
R
90 lines
2.3 KiB
R
|
#' A repeated linear regression function
|
||
|
|
#'
|
||
|
|
#' For bivariate analyses.
|
||
|
|
#' @param y Effect meassure.
|
||
|
|
#' @param v1 Main variable in model
|
||
|
|
#' @keywords linear regression
|
||
|
|
#' @export
|
||
|
|
#' @examples
|
||
|
|
#' rep_lm()
|
||
|
|
|
||
|
|
rep_lm<-function(y,v1,string,ci=FALSE,data,v2=NULL,v3=NULL){
|
||
|
|
## x is data.frame of predictors, y is vector of an aoutcome as a factor
|
||
|
|
## output is returned as coefficient, or if ci=TRUE as coefficient with 95 % CI.
|
||
|
|
## The confint() function is rather slow, causing the whole function to hang when including many predictors and calculating the ORs with CI.
|
||
|
|
|
||
|
|
require(broom)
|
||
|
|
|
||
|
|
d<-data
|
||
|
|
x<-select(d,one_of(c(string)))
|
||
|
|
m1<-length(coef(lm(y~v1)))
|
||
|
|
|
||
|
|
if (is.factor(y)){stop("Some kind of error message would be nice, but y should not be a factor!")}
|
||
|
|
|
||
|
|
if (ci==TRUE){
|
||
|
|
|
||
|
|
df<-data.frame(matrix(ncol = 4))
|
||
|
|
names(df)<-c("pred","co_ci","pv","t")
|
||
|
|
|
||
|
|
for(i in 1:ncol(x)){
|
||
|
|
m<-lm(y~v1+x[,i])
|
||
|
|
|
||
|
|
l<-suppressMessages(round(confint(m)[-c(1:m1),1],2))
|
||
|
|
u<-suppressMessages(round(confint(m)[-c(1:m1),2],2))
|
||
|
|
co<-round(coef(m)[-c(1:m1)],2)
|
||
|
|
|
||
|
|
co_ci<-paste0(co," (",l," to ",u,")")
|
||
|
|
|
||
|
|
pv<-round(tidy(m)$p.value[-c(1:m1)],3)
|
||
|
|
pv<-ifelse(pv<0.001,"<0.001",pv)
|
||
|
|
|
||
|
|
t <- ifelse(pv<=0.1|pv=="<0.001","include","drop")
|
||
|
|
|
||
|
|
pv <- ifelse(pv<=0.05|pv=="<0.001",paste0("*",pv),
|
||
|
|
ifelse(pv>0.05&pv<=0.1,paste0(".",pv),pv))
|
||
|
|
|
||
|
|
|
||
|
|
v<-x[,i]
|
||
|
|
|
||
|
|
if (is.factor(v)){
|
||
|
|
pred<-paste(names(x)[i],levels(v)[-1],sep = "_")
|
||
|
|
}
|
||
|
|
|
||
|
|
else {pred<-names(x)[i]}
|
||
|
|
|
||
|
|
df<-rbind(df,cbind(pred,co_ci,pv,t))
|
||
|
|
|
||
|
|
}}
|
||
|
|
|
||
|
|
if (ci==FALSE){
|
||
|
|
|
||
|
|
df<-data.frame(matrix(ncol = 4))
|
||
|
|
names(df)<-c("pred","b","pv","t")
|
||
|
|
|
||
|
|
for(i in 1:ncol(x)){
|
||
|
|
m<-lm(y~v1+x[,i])
|
||
|
|
|
||
|
|
b<-round(coef(m)[-c(1:m1)],3)
|
||
|
|
|
||
|
|
pv<-round(tidy(m)$p.value[-c(1:m1)],3)
|
||
|
|
pv<-ifelse(pv<0.001,"<0.001",pv)
|
||
|
|
|
||
|
|
t <- ifelse(pv<=0.1|pv=="<0.001","include","drop")
|
||
|
|
|
||
|
|
pv <- ifelse(pv<=0.05|pv=="<0.001",paste0("*",pv),
|
||
|
|
ifelse(pv>0.05&pv<=0.1,paste0(".",pv),pv))
|
||
|
|
|
||
|
|
v<-x[,i]
|
||
|
|
|
||
|
|
if (is.factor(v)){
|
||
|
|
pred<-paste(names(x)[i],levels(v)[-1],sep = "_")
|
||
|
|
}
|
||
|
|
|
||
|
|
else {pred<-names(x)[i]}
|
||
|
|
|
||
|
|
df<-rbind(df,cbind(pred,b,pv,t))
|
||
|
|
}}
|
||
|
|
|
||
|
|
return(df)
|
||
|
|
}
|