如何在列表理解中包含多个要素

如何解决如何在列表理解中包含多个要素

我想将primes生成的五个连续素数作为一个无限列表，并检查它们的总和是否为另一个素数。我想要这样的东西：

consecutivePrimes = [ a+b+c+d+e | a:b:c:d:e <- primes,prime a+b+c+d+e]

这个a:b:c:d:e <- primes不能正常工作，而且我找不到在列表理解中一次获取多个元素的任何方法。

解决方法

由于列表理解可以看作是map与filter结合在一起（至少对于列表而言），因此一次只能在其中包含一个元素。

但是您仍然可以通过使用primes将tails制成列表列表，然后从每个列表中选取5个元素来实现。在这种情况下，您映射到的单个元素（ps是一个列表。

import Data.List (tails)
consecutivePrimes = [ a+b+c+d+e | ps <- tails primes,let [a,b,c,d,e] = take 5 ps,prime a+b+c+d+e]

如果输入列表是无限的，则在5个元素的列表上进行模式匹配将始终成功。

,

这是我的解决方案，但是很丑陋：

consecutivePrimes = [x | x <- consecutivePrimes' primes,prime x]
consecutivePrimes' (a:b:c:d:e:xs) | prime (a+b+c+d+e) = (a+b+c+d+e) : consecutivePrimes' (b:c:d:e:xs)
                                  | otherwise = consecutivePrimes' (b:c:d:e:xs))