haskell - Haskell 中的 groupByKey - 如何通过函数对列表中的项目进行分组？

Question

给定一个列表[v]和一个键控函数f::v -> k，我想生成一个 map Map k [v]。 Haskell 中是否存在这样的东西？

import Data.Map

groupByKey :: (v -> k) -> [v] -> Map k [v]

Answer 1

您可以使用fromListWith，如下所示:

import Data.Map
groupByKey :: (Ord k) => (v -> k) -> [v] -> Map k [v]
groupByKey getkey
  = fromListWith (++) . fmap (\val -> (getkey val, [val]))

这样:

> groupByKey length $ words "hello there my good friend"
fromList [(2,["my"]),(4,["good"]),(5,["there","hello"]),(6,["friend"])]
>