homework for 可爱的python