結果 : taking input in function python