Haskell - 列表理解，创建/扩展元组列表

Question

我正在尝试为元组列表制定紧凑的更新规则。

这是更新规则的示例：

rule1: [(a,b) | (a,b) <- zip [1..3] [2..4]]
λ:[(1,2),(2,3),(3,4)]

我想要的是：

rule2: [(a,b) (b,a) | (a,b) <- zip [1..3] [2..4]]
λ(expected):[(1,2),(2,1),(2,3),(3,2),(3,4),(4,3)]

在 Haskell 的列表理解中，这样的事情可能吗？

---

一个可行的解决方案，但我觉得它有点难看：

concat [[(a,b),(b,a)] | (a,b) <- zip [1..3] [2..4]]
λ: [(1,2),(2,1),(2,3),(3,2),(3,4),(4,3)]

Answer 1

首先，在您的rule1 ，您根本不需要列表理解。 zip的结果已经是你想要的元组列表：

rule1 = zip [1..3] [2..4]

但是，对于rule2 ，您可以只使用两个嵌套列表，第二个从第一个的每个元素派生，如下所示：

rule2 = [p | (a, b) <- zip [1..3] [2..4], p <- [(a, b), (b, a)]]

或者，在不解构元组的情况下，使用swap代替：

rule2 = [p | p0 <- zip [1..3] [2..4], p <- [p0, swap p0]]