educoder 头歌 python 素数写入文件

任务描述

本关任务：编写一个将1-9999之间的素数分别写入三个文件中（1-99之间的素数保存在a.txt中，100-999之间的素数保存在b.txt中，1000-9999之间的素数保存在c.txt中）。

相关知识

为了完成本关任务，你需要掌握：1.文件读写，2.循环语句，3.条件语句，4.函数定义及调用。

编程要求

根据提示，在右侧编辑器补充代码，将1-99之间的素数保存在a.txt中，100-999之间的素数保存在b.txt中，1000-9999之间的素数保存在c.txt中。

测试说明

本次题目不需输入输出，只需将规定的素数保留到规定文件中，平台会编写测试脚本对3个文件进行测试。 a.txt中数据格式为

from math import sqrt


def is_prime(n):
    """判断素数的函数"""
    # assert n > 0
    for factor in range(2, int(sqrt(n)) + 1):
        if n % factor == 0:
            return False
    return True if n != 1 else False



def store():
    #         请在此处添加代码       #
    # *************begin************#
    with open("a.txt",'w') as f:
        for i in range(2,100):
            if is_prime(i):
                f.write(str(i) + "\n")
            
    with open("b.txt",'w') as f:
        for i in range(100,1000):
            if is_prime(i):
                f.write(str(i) + "\n")
            
    with open("c.txt",'w') as f:
        for i in range(1000,10000):
            if is_prime(i):
                f.write(str(i) + "\n")
            
               
    # **************end*************#