Блог

Я пытаюсь создать эту стратифицированную гистограмму на R, однако я не получаю правильный график. Я хотел бы также использовать функцию rect, если это возможно.

Как бы я написал функцию R для создания этой стратифицированной гистограммы с использованием набора данных iris в R?

Это код на данный момент:

 strathist = function(x, y, ylab = "Frequency", xlab = "", main = ""){
  
  cols = hcl(h=seq(0, 360, by = 120))
  
  h = hist(x, breaks = 24, plot = FALSE)
  
  tb = table(y, cut(x, h$breaks))
  
  
  plot.new()
  
  barplot(tb, ylim = c(0, max(h$count)), col = cols,
          ylab = ylab, xlab = xlab, main = main, axisnames = FALSE) 
  
  box()
  axis(1, 0:(length(h$breaks)-1), h$breaks)
  axis(2)
  

  
  legend("topright", c(rownames(tb)), fill = cols)
}

with(iris, strathist(Sepal.Width, Species, xlab = "Sepal.Width", main = "Stratified Histogram"))
  

Есть несколько различных способов сделать это, если вы настаиваете на использовании R base, вы могли бы использовать две гистограммы, одну для добавления цветов, а другую для добавления полей:

 strathist = function(x,
                     y,
                     ylab = "Frequency",
                     xlab = "",
                     main = "") {
  cols = hcl(h = seq(0, 360, by = 120))
  
  h = hist(x,
           breaks = 24,
           plot = F)
  
  tb = table(y, cut(x, h$breaks))
  
  
  ylim <- c(-.05 * max(colSums(tb)),
            1.25 * max(colSums(tb)))
  
  barplot(
    tb,
    col = cols,
    ylim = ylim,
    ylab = ylab,
    xlab = xlab,
    main = main,
    axisnames = FALSE,
    border = NA,
    space = 0
  )
  
  barplot(
    colSums(tb),
    ylim = ylim,
    col = NA,
    ylab = ylab,
    xlab = xlab,
    main = main,
    axisnames = FALSE,
    add = T,
    space = 0
  )
  
  box()
  axis(1, seq(0,
              length(h$breaks) - 1,
              by = 5),
              seq(min(x),max(x), by = .5))
  axis(2)
  
  legend("topright", c(rownames(tb)), fill = cols,border = NA,)
}

with(
  iris,
  strathist(Sepal.Width, Species, xlab = "Sepal.Width", main = "Stratified Histogram")
)
  

Или вы могли бы пойти несколько более простым путем ggplot2 :

 library(ggplot2)

ggplot(iris,aes(x = Sepal.Width))  
  geom_histogram(bins = 25,aes(fill = Species))  
  geom_histogram(bins = 25, fill = NA, color = 'black')  
  theme_minimal()
  

^{Создано 2020-09-13 пакетом reprex (версия 0.3.0)}