Sum over Haskell Map

Question

Hay una función estándar para sumar todos los valores en un mapa Haskell. Mi mapa dice algo como [(a, 2), (b, 4), (c, 6)]?

Esencialmente, lo que estoy tratando de hacer es una distribución de frecuencia normalizada. Entonces, los valores de las claves en el mapa de arriba son cuentas para a, b, c. Necesito normalizarlos como [(a, 1/6), (b, 1/3), (c, 1/2)]

Answer 1

Puede simplemente hacer Map.foldl' (+) 0 (o M.foldl', si importó Data.Map como M).

Esto es como foldl' (+) 0 . Map.elems, pero un poco más eficiente. (No olvides el apóstrofo - usar foldl o foldr para hacer sumas con los tipos numéricos estándar (Int, Integer, Float, Double, etc.) generará enormes thunks, que consumirán mucha memoria y posiblemente causen tu programa para desbordar la pila.)

versiones sin embargo, sólo lo suficientemente recientes de containers (> = 0.4.2.0) contienen Data.Map.foldl', y no debe actualizar con cabal install, ya que viene con GHC. A menos que esté en GHC 7.2 o superior, foldl' (+) 0 . Map.elems es la mejor manera de lograr esto.

También es posible usar Data.Foldable.sum, que funciona en cualquier instancia de la clase de tipos Foldable, pero seguirá siendo acumular grandes procesadores de los tipos numéricos comunes.

Aquí está un ejemplo completo:

normalize :: (Fractional a) => Map k a -> Map k a 
normalize m = Map.map (/ total) m 
    where total = foldl' (+) 0 $ Map.elems m 

Tendrá que importar Data.List utilizar foldl'.

Answer 2

let 
    total = foldr (\(_, n) r -> r + n) 0 l 
in map (\(x, y) -> (x, y/total) l 

Dónde l es su mapa.

Answer 3

simple:

import qualified Data.Map as M 

sumMap = M.foldl' (+) 0 

normalizeMap m = 
    let s = sumMap m in 
    M.map (/ s) m 

main = do 
    let m = M.fromList [("foo", 1), ("bar", 2), ("baz", 6)] 
    (print . sumMap) m 
    (print . normalizeMap) m 

impresiones:

9.0 
fromList [("bar",0.2222222222222222),("baz",0.6666666666666666),("foo",0.1111111111111111)]