結果 : how to write one line for loop in python