有没有更好的方法来完成这个python练习？ （初学者）

I'm just starting to learn Python and I'm going through an exercise at the end of a chapter. So far, all I've learned about in the book is the very basics, flow control, functions, and lists.I'm just starting to learn Python and I'm going