如果谓词成立则映射函数

Question

我觉得这应该是相当明显或容易的，但我就是不明白。我想要做的是将函数应用于列表（使用 map），但前提是满足条件。想象一下，您只想除偶数：

map (`div` 2) (even) [1,2,3,4]

这会给出 [1,1,3,2] ，因为只有偶数才会应用该函数。显然这是行不通的，但是有没有一种方法可以让这个工作不需要编写一个可以提供给 map 的单独函数？ filter 几乎就在那里，除了我还想保留条件不成立的元素，并且只是不将函数应用于它们。

Answer 1

如果您不想定义单独的函数，请使用 lambda。

map (\x -> if (even x) then (x `div` 2) else x) [1,2,3,4]

或者用列表理解代替地图，我认为更具可读性。

[if (even x) then (x `div` 2) else x | x <- [1,2,3,4]]

Answer 2

mapIf p f = map (\x -> if p x then f x else x)

Answer 3

除了 PiotrLegnica 的答案：通常，如果声明一个辅助函数而不是使用 lambda，则更容易阅读。考虑一下：（

map helper [1..4] where
  helper x | even x    = x `div` 2
           | otherwise = x

[1..4] 是 [1,2,3,4] 的糖）

如果您想删除所有其他元素，请考虑使用过滤器。 filter 删除所有不满足谓词的元素：

filter even [1..4] -> [2,4]

因此您可以构建一个 map 管道并过滤或使用列表理解：

map (`div` 2) $ filter even [1..4]
[x `div` 2 | x <- [1..4], even x]

选择您最喜欢的任何内容。

Answer 4

制作你自己的辅助函数生成器：（

ifP pred f x = 
    if pred x then f x
    else x

custom_f = ifP even f
map custom_f [..]

警告：我现在无法访问编译器。我想这可以正常工作......）

Answer 5

我喜欢其他更通用的解决方案，但在你非常特殊的情况下，你可以逃脱

map (\x -> x `div` (2 - x `mod` 2)) [1..4]

Answer 6

大部分是对现有答案的抄袭，但根据我对“可读”的有偏见的定义（我更喜欢守卫而不是 ifs，并且 where 比 let 更喜欢）：

mapIf p f = map f'
    where f' x | p x = f x | otherwise = x

ghci 这么说可能有效

ghci> let mapIf p f = map f' where f' x | p x = f x | otherwise = x
ghci> mapIf even (+1) [1..10]
[1,3,3,5,5,7,7,9,9,11]