List Comprehensions
In this chapter, we introduce list comprehensions, which are used to create new lists from existing ones. The term comes from mathematics, where set comprehension is used to describe a set by enumerating its elements and/or stating conditions its members must satisfy. In Haskell, the syntax is the following:
[ <GENERATOR> | <ELEMENT> <- <LIST>, <GUARD> ]
The GENERATOR is an expression that specifies how the elements of the new lists should be calculated, the ELEMENT is an element from the specified LIST, and the GUARD is an optional condition we can set that an ELEMENT must satisfy to be considered for the new list we are creating. The whole line can be read as "create a list by applying the GENERATOR for each ELEMENT of the LIST that meets the criteria set by the GUARD". Let's take a look at an example list comprehension that creates a list of only even numbers:
ghci> [x | x <- [1..10], even x]
[2,4,6,8,10]
The above can be read as "create a list of all numbers x such that x is an element of the list [1..10] and x is an even number". This simply puts x into a new list if it meets the guard criteria, but we could also apply some function to x:
ghci> [x * 2 | x <- [1..10], even x]
[4,8,12,16,20]
Or not even use x as it is in the list for our generator, but just use it to make sure it passes the guards check:
ghci> ["even!" | x <- [1..10], even x]
["even!","even!","even!","even!","even!"]
We can also specify multiple lists as well as multiple guards in list comprehensions. In the case of multiple lists, the deeper nested list (the one specified last) is iterated through for each element of the first list:
ghci> [ (x, y) | x <- [1..3], y <- ['a'..'c'] ]
[(1,'a'),(1,'b'),(1,'c'),(2,'a'),(2,'b'),(2,'c'),(3,'a'),(3,'b'),(3,'c')]
In this case, the list ['a'..'b'] is iterated through three times, once for each element in the first list [1..3].
Copy link