ML作业的python代码