Handle either a list or single integer as an argument

Question

A function should select rows in a table based on the row name (column 2 in this case). It should be able to take either a single name or a list of names as arguments and ha

Answer 1

A simple wrapper could do, to handle list or single object case

  def wrap_list(val):
    if type(val) is list:
      return val
    return [val]