繁体   English   中英

使用lambda函数转换double for循环并中断对列表的理解

[英]convert double for loop with lambda function and break to list comprehension

我遵循以下代码,我需要将其转换为list-comprehension (一种衬里)。 但是,我无法这样做。 该代码计算素数直至输入A范围。

def sieve(A):    
    l = []
    f = lambda x : int(x**0.5)
    for p in range(2,A+1):        
        for i in range(2, f(p) + 1):
            if p % i == 0:
                break
        else:
            l.append(p)
    return l

到目前为止,我一直关注的是行不通的。 尤其是for-loop内的break for-loop我无法接受。

list(set([val for sublist in [[p  for i in range(2, f(p) + 1) if p %i != 0 ] for p in range(2,A) ] for val in sublist]))

编辑
为问题添加约束。 该代码只能是一个语句,不能包含evalexec 该代码最多不能超过160个字符。

[p for p in range(2,A+1) if next((i for i in range(2, int(p**0.5) + 1) if (p % i) == 0),None)==None]

代码长度为100个字符。
我们使用next()打破迭代。
说明

def sieve(A):
  [p for p in range(2,A+1) if getFirstDiv(p)==None]

def getFirstDiv(p):
  next(divIter(p),None)

def divIter(p):
  return (i for i in range(2, int(p**0.5) + 1) if (p % i) == 0)

OUTPUT

15 --> [2, 3, 5, 7, 11, 13]
10 --> [2, 3, 5, 7]

这一个班轮将做到这一点:

[r for r in [i*all([i if i%j!=0 else 0 for j in range(2,i)]) for i in range(2,x)] if r>0]

您只需要设置x (最大值)即可。

注意:虽然我猜效率不是这个问题的目的,但它并不是特别有效。

说明(扩展代码):

filtered = []
primes = []

for i in range(2,x):
    # Check that all numbers up to i do not divide i
    # I realise we only need to check up to int(sqrt(i))
    condition = all([i if i%j!=0 else 0 for j in range(2,i)])

    # Exploit Python's treatment of bool: number*True = number and number*False=0
    filtered.append( i*condition )


for r in filtered:
    # Take out all the zeros
    if r>0:
        primes.append(r)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM