Haskell-如何在列表理解中使用if else

Question

我希望通過擲兩個骰子來生成點數列表。 如果兩個骰子顯示相同的值，則輸出點將變為(i,i,i,i) 。

我的代碼就像

[(i,j)|i<-[1..6],j<-[1..6], if i==j
                            then (i,i,i,i)
                            else (i,j)]

當我將其輸入到GHCi時，無法對其進行編譯。

我期望的輸出是

[(1,1,1,1),(1,2),(1,3),(1,4),(1,5),(1,6),(2,1),(2,2,2,2),(2,3),(2,4),(2,5),(2,6),(3,1),(3,2),(3,3,3,3),(3,4),(3,5),(3,6),(4,1),(4,2),(4,3),(4,4,4,4),(4,5),(4,6),(5,1),(5,2),(5,3),(5,4),(5,5,5,5),(5,6),(6,1),(6,2),(6,3),(6,4),(6,5),(6,6,6,6)]

Answer 1

列表元素應該是相同的類型，而不是您的4元組和2元組。 如果您切換到列表，則可以這樣寫

[if i==j then [i,i,i,i] else [i,j] | i<-[1..6], j<-[1..6]]

會產生

[[1,1,1,1],[1,2],[1,3],[1,4],[1,5],[1,6],
 [2,1],[2,2,2,2],[2,3],[2,4],[2,5],[2,6],
 [3,1],[3,2],[3,3,3,3],[3,4],[3,5],[3,6],
 [4,1],[4,2],[4,3],[4,4,4,4],[4,5],[4,6],
 [5,1],[5,2],[5,3],[5,4],[5,5,5,5],[5,6],
 [6,1],[6,2],[6,3],[6,4],[6,5],[6,6,6,6]]