Python新手的作业练习

有道作业如下：
写代码计算1至10十个整数的平方，如果平方的结果大于等于50，保存到sq_50.txt中；如果平方的结果小于50，保存到sq_49.txt中。

自己写的代码如下：

file1=open('D:/桌面/sq_50.txt','w')
file2=open('D:/桌面/sq_49.txt','w')
n=range(1,11)
for i in n:
    sqr=i**2
    if sqr>=50:
        file1.write(str(sqr)+'\n')
    else:
        file2.write(str(sqr)+'\n')
file1.close()
file2.close()

下图为两个文本文档的内容：