Question

In haskell I can use

sortBy (comparing snd) 

to sort by the second value in a tuple.

Is there an equivalent function for testing equivalency? I've come up with this but maybe there is something in the standard library.

equalsBy :: Eq b => (a -> b) -> a -> a -> Bool
equalsBy f x y = f x == f y

The end goal is to group a list of pairs by their second values. With this I can do

groupBy (equalsBy snd) pairs

instead of

groupBy (\x y -> (snd x) == (snd y)) pairs
Was it helpful?

Solution

You are looking for a slightly higher level function called on:

> import Data.Function
> :t on
on :: (b -> b -> c) -> (a -> b) -> a -> a -> c

Well how do I use that? Like this!

> :t ( (==) `on` snd )
( (==) `on` snd ) :: Eq b => (a, b) -> (a, b) -> Bool

So that means we want:

> groupBy ( (==) `on` snd) [ (1,3), (23,9), (42,9), (1,3), (48, 3), (18,18)]
[[(1,3)],[(23,9),(42,9)],[(1,3),(48,3)],[(18,18)]]

Yay!

EDIT:

I'd like to note how comparing relates to on. It is just a specialized use of on!

> :t comparing
comparing      :: Ord a => (b -> a) -> b -> b -> Ordering
> :t (compare `on`)
(compare `on`) :: Ord b => (a -> b) -> a -> a -> Ordering

(notice the type variables are swapped, but you can see the types are identical)

Licensed under: CC-BY-SA with attribution
Not affiliated with StackOverflow
scroll top