添加assignment1.py文件，这个是第一套题目的参考