[R 통계_ggplot2 패키지] 01-2. geom_point에 stat_function으로 맞춤형 식 추가하기

이전 포스팅에서는 ggplot2 패키지에서 geom_point로 산점도를 그린 후,

geom_abline을 통해, 회귀식을 삽입해보았다.

 

이번엔, stat_function 기능을 통해 맞춤형 함수식을 삽입해보자

 

03-1. stat_function 추가하기

그래프1 <- ggplot(데이터명, aes(x축, y축))
그래프1 <- 그래프1 + geom_point(colour = "색상", size = 숫자)
그래프1 <- 그래프1 + stat_function(data = 데이터명, fun = 맞춤 함수식 명, colour = "색상", size = 숫자)

 

우선, 임의의 곡선 데이터를 제작해보았다.

# 2차 곡선형태의 데이터 산출
x <- seq(1,20)
y <- c(2, 4, 5, 6, 6.5, 6, 7, 7.5, 6.8, 7.2, 7.3, 7.6, 7.5, 7, 6.4, 5.5, 5.6, 4.8, 3, 1)
my_data <- data.frame(x, y)

엑셀에서 대략적인 추세를 보니, 다음과 같이 2차식 곡선형태의 데이터가 산출되었는데,

추정식은 다음과 같았다. 

y = -0.0597*x^2 + 1.2158*x + 1.4792

excel을 활용한 데이터 추세식 산출

 

이러한 정보를 바탕으로 맞춤형 함수를 제작하고,

geom_point와 stat_function을 통해 맞춤형 수식 Fitting을 진행하였다.

f <- function(x){
  return(-0.0597*x^2+1.2158*x+1.4792)
  }
  
library(ggplot2)

#03-1. stat_function을 통해 맞춤 회귀선 추가하기
point <- ggplot(my_data, aes(x, y))
point <- point + geom_point(colour = "blue", size =2)
point <- point + stat_function(data = my_data, fun = f, colour = "blue", size = 1.2)
point

geom_point with stat_function()