En utilisant foldl pour compter le nombre de vraies valeurs

Question

J'essaie de trouver un moyen de faire la fonction suivante avec foldl:

count a = length (filter (\i -> i) a) 

Il compte que le nombre de valeurs qui sont vraies dans une liste de booléens. Je l'ai essayé moi-même avec

count = foldl (\i -> 
      case i of 
       True -> (1+) 
       False -> (0+) 
      ) 0 

Qui n'a même pas compilé. Aucune suggestion?

Answer 1

Alors nous allons vérifier les types de fonctions impliquées

Prelude> :t (\i -> case i of { True -> (1+) ; False -> (0+) }) 
(\i -> case i of { True -> (1+) ; False -> (0+) }) :: (Num t) => Bool -> t -> t 
Prelude> :t foldl 
foldl :: (a -> b -> a) -> a -> [b] -> a 

Donc, pour votre liste de Bool s, b est Bool, mais la fonction que vous utilisez a Bool comme premier argument, et non la seconde . La valeur accumulée est le premier argument. Ainsi, au lieu que vous pourriez faire

foldl (\acc p -> case p of { True -> acc + 1 ; False -> acc }) 0 

Ou si vous souhaitez simplement fixer l'ordre des arguments, utilisez votre fonction originale avec flip

Prelude> :t flip 
flip :: (a -> b -> c) -> b -> a -> c 

foldl (flip (\i -> case i of 
          True -> (1+) 
          False -> (0+) 
      )) 0 

Ou vous pouvez être plus succincte: foldl (flip ((+) . fromEnum)) 0

Answer 2

Et à propos de:

count = foldl (\i v -> if v then i + 1 else i) 0 

Une autre façon de le faire sans t foldl:

count list = sum $ map fromEnum list 

Crédit Logan pour souligner fromEnum. N'avait pas entendu parler de celui-là avant.