Замена вложенного на l oop, который генерирует цифры с применением семейства - PullRequest
Новая
       32

Замена вложенного на l oop, который генерирует цифры с применением семейства

0 голосов
/ 27 апреля 2020

Фон

У меня есть количественный набор данных из опроса. Я хочу построить график распределения tri angular для имеющихся у меня значений (минимум lb, максимум ub и режим ml). Имейте в виду, я использую rtriang(), так как мои данные не содержат квантилей, для которых может быть установлена ​​функция плотности. По крайней мере, это мое понимание.

Вопрос

  • Сейчас я использую уродливое гнездо для l oop, и, вероятно, это более эффективно сделать с помощью apply-family функций, хотя я не знаю, как это сделать. Как бы я это сделал?

Код 

library(data.table)
library(ggplot2)
library(mc2d)

scenarios <- c("s1", "s2")
questions <- c("q1", "q2")
respondents <- c("1","2","3")

data_long <- data.frame(id=c("1","2","3", "1","2","3", "1","2","3",
                               "1","2","3", "1","2","3", "1","2","3",
                               "1","2","3", "1","2","3", "1","2","3",
                               "1","2","3", "1","2","3", "1","2","3"),
                         variable=c("s1_q1_ml", "s1_q1_ml", "s1_q1_ml",
                                      "s1_q1_lb", "s1_q1_lb", "s1_q1_lb",
                                      "s1_q1_ub", "s1_q1_ub", "s1_q1_ub",
                                      "s1_q2_ml", "s1_q2_ml", "s1_q2_ml",
                                      "s1_q2_lb", "s1_q2_lb", "s1_q2_lb",
                                      "s1_q2_ub", "s1_q2_ub", "s1_q2_ub",
                                      "s2_q1_ml", "s2_q1_ml", "s2_q1_ml",
                                      "s2_q1_lb", "s2_q1_lb", "s2_q1_lb",
                                      "s2_q1_ub", "s2_q1_ub", "s2_q1_ub",
                                      "s2_q2_ml", "s2_q2_ml", "s2_q1_ml",
                                      "s2_q2_lb", "s2_q2_lb", "s2_q1_lb",
                                      "s2_q2_ub", "s2_q2_ub", "s2_q1_ub"),
                         value=c(70, 70, 70, 60, 60, 60, 80, 80, 80,
                                   70, 70, 70, 60, 60, 60, 80, 80, 80,
                                   70, 70, 70, 60, 60, 60, 80, 80, 80,
                                   70, 70, 70, 60, 60, 60, 80, 80, 80))

data_long <- setDT(data_long)

for (i in respondents) {
  for (j in scenarios) {
    for (k in questions) {
      t <- rtriang(n =100000, min=as.numeric(data_long[id==i & variable == paste(j, k, "lb", sep = "_")]$value), 
                   mode=as.numeric(data_long[id==i & variable == paste(j,k, "ml", sep = "_")]$value),
                   max=as.numeric(data_long[id==i & variable == paste(j,k, "ub", sep = "_")]$value))

      # Displaying the samples in a density plot
      plot <- ggplot() + geom_density(aes(t)) + xlim(0,100) + xlab("Probability in %")
      ggsave(plot,filename=paste(i,j,k,".png",sep="_"))
    }
  }
}

1 Ответ

1 голос
/ 27 апреля 2020

A tidyverse подход: 

library(tidyverse)
library(mc2d)

all_plots <- data_long %>%
               separate(variable, c("scenarios", "questions", "temp"),
                         sep = "_") %>% 
               group_split(id, scenarios, questions) %>%
               map(~{
                    temp <- rtriang(
                      n =100000, 
                      min = .x %>% filter(temp == 'lb') %>% pull(value),
                      mode = .x %>% filter(temp == 'ml') %>% pull(value),
                      max = .x %>% filter(temp == 'ub') %>% pull(value))
                      plot <- ggplot() + 
                               geom_density(aes(temp)) + xlim(0,100) + 
                               xlab("Probability in %")
                       ggsave(filename = paste(.x$id[1],.x$scenarios[1],
                                        .x$questions[1],".png",sep="_"), plot)
                   })
...