refactoring code with List Comprehension

[idiomaticrefactoring]

refactoring code with List Comprehension which is more pythonic, concise and efficient; how do you think this change which has practical value？

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}

[zhangjiajin]

Good。 @zjzh