Concise way of checking arguments with python list comprehension feature