Haskell实现map后过滤Nothing结果的自定义函数问题咨询

代码错误原因 #

• 语法错误
  你当前的mapThenFilter定义没有声明参数，且函数应用的括号位置错误。Haskell中函数参数直接跟在函数名后，不需要括号包裹，你写的map checkIfNothing(map f l)会被解析为非法的多参数调用，正确的基础写法应该先声明参数mapThenFilter f l = ...，函数组合需要显式加括号：map checkIfNothing (map f l)。
• 核心逻辑错误
  你定义的checkIfNothing函数是把Nothing替换为0，而非丢弃对应元素。map操作是一对一的等长转换，输入列表有多少个元素，输出就有多少个元素，完全不可能实现「丢弃元素」的效果，你当前的代码只会把所有Nothing结果替换为0保留在列表中，和需求完全不符。
• 对filter的作用理解偏差
  你需要的操作分为三步：
  1. 用传入的函数f映射原始列表，得到类型为[Maybe Integer]的中间列表
  2. 过滤掉中间列表里所有的Nothing元素
  3. 提取剩余Just包裹的整数值，得到最终结果
     这三步里第二步必须用到筛选类操作，map无法单独完成。

正确实现 #

写法1：基础map+filter组合（不用高阶工具函数） #

mapThenFilter :: (Integer -> Maybe Integer) -> [Integer] -> [Integer]
mapThenFilter f l = map (\(Just x) -> x) (filter (\v -> case v of Just _ -> True; Nothing -> False) (map f l))

也可以把判断逻辑拆分得更清晰：

isJust :: Maybe a -> Bool
isJust (Just _) = True
isJust Nothing = False

fromJust :: Maybe a -> a
fromJust (Just x) = x

mapThenFilter f l = map fromJust (filter isJust (map f l))

写法2：列表推导式（更简洁） #

mapThenFilter :: (Integer -> Maybe Integer) -> [Integer] -> [Integer]
mapThenFilter f l = [x | num <- l, Just x <- [f num]]

这个写法利用了Haskell列表推导的模式匹配特性，f num匹配不上Just x的元素会被自动丢弃，刚好符合你的需求。

写法3：使用标准库mapMaybe #

Data.Maybe模块内置的mapMaybe函数刚好就是实现这个逻辑的，一行即可完成：

import Data.Maybe (mapMaybe)

mapThenFilter :: (Integer -> Maybe Integer) -> [Integer] -> [Integer]
mapThenFilter = mapMaybe

内容的提问来源于stack exchange，提问作者user202004