python txt文件操作

最新推荐文章于 2023-04-10 11:42:35 发布

wbl320

最新推荐文章于 2023-04-10 11:42:35 发布

阅读量142

点赞数 2

本文链接：https://blog.csdn.net/wbl320/article/details/96478116

版权

将10000以内的素数分别写入txt中

from math import sqrt


def prime(n):
    for i in range(2,int(sqrt(n))+1):
        if n % i == 0:
            return False
    return True


def main():
    file_name = ['a.txt', 'b.txt', 'c.txt']
    file = []
    #f1 = open(file_name[0], 'w')
    #f2 = open(file_name[1], 'w')
    #f3 = open(file_name[2], 'w')
    try:
        for i in file_name:
            file.append(open(i, 'w'))
        for i in range(2, 10000):
            if prime(i):
                if i < 99:
                    file[0].write(str(i) + '\n')
                elif i < 999:
                    file[1].write(str(i) + '\n')
                else:
                    file[2].write(str(i) + '\n')
    except IOError as ex:
        print(ex)
        print('写入文件发生错误')
    finally:
        for i in file:
            i.close()
    print('写入完成！')

if __name__ == '__main__':
    main()