输出质数的一种方式和filter函数

2020-12-06  本文已影响0人  万年星河

先创建相应的输出质数的函数,这里返回的是bool值,便于用filter函数,当然还可以返回函数本身的值

from math import sqrt

def isprime_num(n:int) -> bool:

    if n < 2:

        return False

    for i in range(2, int(sqrt(n)+1)):

        if n % i == 0:

            return False

    return True

定义好函数之后就可以调用函数进行输出质数了,这里是输出50以内的质数

L = [i for i in range (0, 50)] 

print(f'prime num is:{list(filter(isprime_num, L))}')

也可以用这种方式直接输出能被2和3整除的数

L = [i for i in range (0, 50)]

print(f'i%2,i%3==0 num is:{list(filter(lambda x: True if x % 2 == 0 and x%3 == 0 else False, L))}')

filter函数有两个参数,第一参数为你所要用到的函数,第二个为传入到函数的参数(一个序列),序列的值依次放入函数中,为True则将值取出,放入新列表中。

也可以用列表的推导输出奇书和偶数

L = [i for i in range (0, 50)] 

print(f'odd num is: {str([i for i in L if i%2 !=0])}')

print(f'even num is:{str([i for i in L if i%2 == 0])}')

print(f'prime num is:{list(filter(isprime_num, L))}')

print(f'i%2,i%3==0 num is:{str([i for i in L if i%2==0 and i%3 ==0])}')

print(f'i%2,i%3==0 num is:{list(filter(lambda x: True if x % 2 == 0 and x%3 == 0 else False, L))}')

代码以及结果为:

上一篇下一篇

猜你喜欢

热点阅读