haskell - mapMaybe 用于 Data.Map 中的键

Question

我想将以下函数映射到 Map 的键上

f :: a -> Maybe b

并丢弃 Nothing键并保留 Just key ，但从 Just 中提取.就像 Map.mapMaybe , 但对于键

mapMaybeKeys :: (a -> Maybe b) -> Map a c -> Map b c

我在 Hoogle 中搜索了这种类型的签名，但没有找到任何东西。
我可以这样做:

mapMaybeKeys f
    = Map.toList
    . catMaybes
    . fmap (fmap swap . traverse f . swap)
    . Map.toList

或者:

mapMaybeKeys f
   = Map.mapKeys fromJust
   . Map.delete Nothing
   . Map.mapKeys f

有没有更优雅的方式？

Answer 1

通过列表理解，

import Data.Map (Map)
import qualified Data.Map as M
import Control.Arrow (first)

mapMaybeKeys :: Ord b => (a -> Maybe b) -> Map a c -> Map b c
mapMaybeKeys f m = 
   M.fromList [ (b,a) | (Just b, a) <- map (first f) . M.toList $ m ]