Haskell - List(算術的平均値, 中央値, 最頻値を返す関数, Data.Mapモジュール)

すごいHaskellたのしく学ぼう!
(オーム社)
Miran Lipovača (著)
田中 英行 村主 崇行 (翻訳)
[Kindle版]

開発環境

• OS X Mavericks - Apple(OS)
• BBEdit - Bare Bones Software, Inc., Emacs (Text Editor)
• Haskell (純粋関数型プログラミング言語)
• GHC (The Glasgow Haskell Compiler) (処理系)
• The Haskell Platform (インストール方法、モジュール等)

初めてのコンピュータサイエンス(Jennifer Campbell、Paul Gries、Jason Montojo、Greg Wilson(著)長尾 高弘(翻訳))の12章(各種ツール)、12.7(練習問題)、12-17.をHaskellで解いてみる。

その他参考書籍

• プログラミングHaskell (オーム社) Graham Hutton(著) 山本 和彦(翻訳)
• Real World Haskell―実戦で学ぶ関数型言語プログラミング (オライリージャパン) Bryan O'Sullivan John Goerzen Don Stewart(著) 山下 伸夫 伊東 勝利 株式会社タイムインターメディア(翻訳)

12.7(練習問題)、12-17.

コード(BBEdit)

Sample.hs

{-# OPTIONS -Wall -Werror #-}
import qualified Data.List as L
import qualified Data.Map as Map

main :: IO ()
main = do
    print nums1
    putStrLn $ "平均: " ++ (show $ mean $ map fromIntegral nums1)
    print nums2
    putStrLn $ "中央値: " ++ (show $ median $ map fromIntegral nums2)
    print nums3
    putStrLn $ "中央値: " ++ (show $ median$ map fromIntegral nums3)
    print nums4
    putStrLn $ "最頻値: " ++ (show $ mode nums4)
    
mean :: [Double] -> Double
mean ns = (sum ns) / (fromIntegral $ length ns)

median :: [Double] -> Double
median [] = 0
median ns =
    let l = length ns
        n = div l 2
        ms = L.sort ns
    in if mod l 2 == 0 then
        ((ms !! (n - 1)) + (ms !! n)) / 2
    else
        (ms !! n)

mode :: [Int] -> [Int]
mode [] = []
mode ns =
    let l = Map.fromListWith (\n m -> n + m) $ map (\x -> (x, (1::Double))) ns
        a = foldr1 (\x acc -> if x > acc then x else acc) $ Map.elems l
    in Map.keys $ Map.filter (==a) l

f :: Maybe a -> a
f Nothing = undefined
f (Just a) = a

nums1 :: [Int]
nums1 = [1..10]

nums2 :: [Int]
nums2 = [5, 1, 4, 2, 3]

nums3 :: [Int]
nums3 = [5, 1, 4, 2, 3, 6]

nums4 :: [Int]
nums4 = [0, 1, 0, 3, 2, 4, 1]

入出力結果(Terminal, runghc)

$ runghc Sample.hs
[1,2,3,4,5,6,7,8,9,10]
平均: 5.5
[5,1,4,2,3]
中央値: 3.0
[5,1,4,2,3,6]
中央値: 3.5
[0,1,0,3,2,4,1]
最頻値: [0,1]
$

慣れるまでは{-# OPTIONS -Wall -Werror #-}の記述を消さずに細かく型を指定していくことに。