Генерация x случайных чисел из n групп чисел - PullRequest
Новая
       17

Генерация x случайных чисел из n групп чисел

0 голосов
/ 01 сентября 2018

Если у меня есть следующие списки чисел в R, как я могу выбрать 6 случайных чисел, указав, сколько мне нужно от каждой группы? 

r<-c(1,2,3,4,5,6,7,8,9,10)
b<-c(11,12,13,14,15,16,17,18,19,20)
y<-c(21,22,23,24,25,26,27,28,29,30)
g<-c(31,32,33,34,35,36,37,38,39,40)

Как выбрать, скажем, 2 случайных числа из r, 2 случайных числа из b, 1 случайное число из y и 1 случайное число из g?

Мне нужно, чтобы мои коды были достаточно гибкими, чтобы вручную вводить, сколько номеров мне нужно из каждого списка. Я также хочу, чтобы конечный результат был просто списком из 6 случайных чисел, основанных на моем выборе.

1 Ответ

0 голосов
/ 01 сентября 2018

Поместите векторы в один список 

foo <- list(
    c(1,2,3,4,5,6,7,8,9,10),
    c(11,12,13,14,15,16,17,18,19,20),
    c(21,22,23,24,25,26,27,28,29,30),
    c(31,32,33,34,35,36,37,38,39,40)
)

Образец х раз

Если вы не возражаете против замены, то сгенерируйте еще один вектор ничьих (указывает, из какой записи списка выбрать) и выполните итерацию по этому вектору. 

bar <- c(1, 1, 2, 2, 3, 4)
sapply(bar, function(x) sample(foo[[x]], 1))

Пример n раз

Если вы не хотите замену, то вам нужно пробовать только один раз из каждой записи списка. Для этого вам нужно присвоить именам в вашем списке желаемое количество розыгрышей и выполнить итерацию по списку (для каждой записи извлеките имя и сэмплируйте это число). 

names(foo) <- c(2, 2, 1, 1)
unlist(sapply(seq_along(foo), function(x) sample(foo[[x]], names(foo[x]))))

образец 1 раз

Другое решение состоит в том, чтобы сгенерировать вектор вероятностей и передать его как аргумент prob в sample. 

sample(unlist(foo), length(bar),
       prob = rep(table(bar) / length(bar) / lengths(foo), lengths(foo)))

Объяснение:

Сколько образцов из каждой записи нам нужно: table(bar)

1 2 3 4 
2 2 1 1

Назначьте каждой записи вероятность: table(bar) / length(bar)

        1         2         3         4 
0.3333333 0.3333333 0.1666667 0.1666667

Дайте каждому числу, для которого существует вероятность отбора: rep(table(bar) / length(bar) / lengths(foo), lengths(foo))

...

         2          2          2          2          2          2          2          2          2          2 
0.03333333 0.03333333 0.03333333 0.03333333 0.03333333 0.03333333 0.03333333 0.03333333 0.03333333 0.03333333 
         3          3          3          3          3          3          3          3          3          3 
0.01666667 0.01666667 0.01666667 0.01666667 0.01666667 0.01666667 0.01666667 0.01666667 0.01666667 0.01666667 

...
...