因此，例如，假设我有一个数字列表，并且我想创建一个列表，其中包含每个数字乘以 2 和 3。有没有办法做类似下面的事情，但取回一个数字列表而不是数字列表的列表？
mult_nums = [ [(n*2),(n*3)] | n <- [1..5]]
-- this returns [[2,3],[4,6],[6,9],[8,12],[10,15]]
-- but we want [2,3,4,6,6,9,8,12,10,15]
Paul Wicks
问问题
522 次
3 回答
17
我发现扩展列表理解使这更容易阅读：
[ m | n <- [1..5], m <- [2*n,3*n] ]
准确检查它的作用以及它与其他解决方案的关系可能会有所帮助。让我们将其定义为一个函数：
mult lst = [ m | n <- lst, m <- [2*n,3*n] ]
在时尚之后，这脱糖到
mult' lst =
concatMap (\n -> concatMap (\m -> [m]) [2*n,3*n]) lst
表达式concatMap (\m -> [m])被包裹m在一个列表中以便立即将其展平——它相当于map id.
将此与@FunctorSalad 的答案进行比较：
mult1 lst = concatMap (\n -> [n*2,n*3]) lst
我们已经优化了concatMap (\m -> [m])。
现在@vili 的回答是：
mult2 lst = concat [ [(n*2),(n*3)] | n <- lst]
这对：
mult2' lst = concat (concatMap (\n -> [[2*n,3*n]]) lst)
与上面的第一个解决方案一样，我们不必要地创建了一个必须删除的列表列表concat。
我认为没有使用列表推导的解决方案，但对mult1. 我的直觉是，Haskell 编译器通常足够聪明，这无关紧要（或者，concat由于惰性评估，不必要的 s 很便宜（而它们在急切的语言中是致命的））。
于 2009-02-20T03:40:59.027 回答
13
你可以使用连接。
concat [ [(n*2),(n*3)] | n <- [1..5]]
output: [2,3,4,6,6,9,8,12,10,15]
于 2009-02-20T00:25:29.167 回答
5
在某些类似的情况下， concatMap 也可以很方便，尽管在这里它并没有太大变化：
concatMap (\n -> [n*2,n*3]) [1..5]
于 2009-02-20T00:30:42.557 回答