Fish Wang
says to
YSITD
我覺得你可以用個generator之類寫，會比較省事