Lambda With If Statement Python