Code a function, using list comprehension, such that it takes two inputs:
a 2-dimensional array of any size
a list of column indexes