python exercise function之yield生成器