Python One Liner If