[英]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]))
编辑
为问题添加约束。 该代码只能是一个语句,不能包含eval
或exec
。 该代码最多不能超过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.