Пример произвольного количества чисел из списка Haskell

Question

Я начинаю изучать Haskell с фона Ruby. Я ищу, чтобы быть в состоянии получить произвольное количество элементов из списка:

sample [1,2,3,4,5,6,7,8,9,10] 
=> 7 
sample 3 [1,2,3,4,5,6,7,8,9,10] 
=> [4,2,9] 

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

Answer 1

на основе кода в http://ruby-doc.org/core-2.0/Array.html для образца, который выбирает п случайные индексы в массиве, я придумал следующее:

import System.Random 
import Data.List 
import Control.Applicative 

sample1 xs = do 
    let l = length xs - 1 
    idx <- randomRIO (0, l) 
    return $ xs !! idx 

sample 0 xs = return [] 
sample n xs = do 
    let l = min n (length xs) 
    val <- sample1 xs 
    (:) <$> (pure val) <*> (sample (l-1) (delete val xs)) 

В качестве альтернативы, вы можете использовать Control.Monad вместо Control.Applicative и liftM2 (:) (return val) (sample (ct-1) (delete val xs))

Используя delete, накладывается ограничение Eq на тип элементов списка, поэтому вам нужно будет разделить/слить индекс, если вам нужно этого избежать.

Answer 2

Вы можете использовать пакет random.shuffle, который поставляется с функцией shuffle'. Но вам нужен генератор случайных чисел.

Вы можете также искать дополнительные пояснения на Haskell вики: http://www.haskell.org/haskellwiki/Random_shuffle

После того, как ваш список перемешивается, вы можете take n ее элементы.

Answer 3

Код:

import System.Random 

sample :: Int -> [a] -> IO [a] 
sample count lst = go count lst [] 
    where go 0 _ acc = return acc 
     go count xs acc = do 
      rnd <- randomIO :: IO Int 
      let ind = rnd `rem` (length xs) 
       (beg, (r:rs)) = splitAt ind xs 
      go (count-1) (beg ++ rs) (r : acc) 

Поскольку генерация случайных чисел нечисто вы должны быть в монаде IO, хотя вы могли бы генерировать случайные семена, а затем па, что функции. Этот код получает случайный int, поэтому он находится внутри границ списка. Затем разбивает список и возвращает это число, удаляя его из списка, а затем рекурсируя, пока не потребуется больше чисел.