Как построить взвешенное лессовое сглаживание в ggplot2?

Как добавить сглаживание лёсса, которое учитывает вес другого столбца?

Допустим, у меня есть следующий data.frame:

    library(ggplot2)

    df <- data.frame(x=seq(1:21))
    df$y <- df$x*0.3 + 10
    df$weight <- 10

    df[6,] <- c(6, 0.1, 1)
    df[7,] <- c(7, 0.1, 1)

    df[13,] <- c(13, 0.1, 1)
    df[14,] <- c(14, 0.1, 1)

    df[20,] <- c(20, 0.1, 1)
    df[21,] <- c(21, 0.1, 1)

    ggplot(data=df, aes(x=x, y=y, size=weight)) + geom_point() + geom_smooth(method=loess, legend=FALSE)

Построение лесс-сглаживания приводит к следующему:

лесс-сглаживание без весов

Но я хочу использовать вес столбца так, чтобы лесс был таким же, как если бы каждая точка с весом 10 существовала в 10 раз:

    df2 <- subset(df, !(x %in% c(6,7,13,14,20,21)))
    df2$weight <- 1

    df3 <- df2
    for(i in seq(1:9)){
      df3 <- rbind(df3, df2)
    }
    df3 <- rbind(df3, c(6, 0.1, 1))
    df3 <- rbind(df3, c(7, 0.1, 1))

    df3 <- rbind(df3, c(13, 0.1, 1))
    df3 <- rbind(df3, c(14, 0.1, 1))

    df3 <- rbind(df3, c(20, 0.1, 1))
    df3 <- rbind(df3, c(21, 0.1, 1))

    ggplot(data=df3, aes(x=x, y=y)) + geom_point() + geom_smooth(method=loess, legend=FALSE) 
    # or to demonstrate
    ggplot(data=df3, aes(x=x, y=y)) + geom_jitter() + geom_smooth(method=loess, legend=FALSE)

сглаживание с весами

Я нашел вес параметров параметра лесса. Но я не знаю, как вызвать его из geom_smooth().

1 ответ

Решение

Я думаю, мне нужно добавить только другой параметр для AES:

  ggplot(data=df, aes(x=x, y=y, size=weight, weight=weight)) + geom_point() + geom_smooth(method=loess, legend=FALSE)

введите описание изображения здесь

Другие вопросы по тегам