haskell 通过函数检查相等性

Question

在 haskell 中我可以使用

sortBy (comparing snd) 

按元组中的第二个值排序。

是否有用于测试等效性的等效函数？我已经想出了这个，但也许标准库中有一些东西。

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

最终目标是按第二个值对一系列对进行分组。有了这个我就可以做到

groupBy (equalsBy snd) pairs

而不是

groupBy (\x y -> (snd x) == (snd y)) pairs

Answer 1

您正在寻找一个稍高级别的函数，名为 on :

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

那么我该如何使用它呢？像这样!

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

这意味着我们想要:

> 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)]]

耶!

编辑:

我想说明一下比较与on之间的关系。它只是 on 的特殊用法!

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

(注意类型变量已交换，但您可以看到类型是相同的)