Как реализовать двумерный массив, чтобы мы могли легко указать, какие столбцы нам нужны?

Question

mlb_data = pd.read_csv(file)

melbourne_features = ['Rooms', 'Bathroom', 'Landsize', 'Lattitude', 'Longtitude']

X = mlb_data[melbourne_features]

X теперь будет содержать только те столбцы, которые нам нужны (указано в melbourne_features). Как это реализовано? Я пытался использовать словарь, 2d массив, наборы. Но, похоже, ничего не работает.

редактировать: пробовать со словарями:

>>> data = {'col1': 'answ1', 'col2': 'answ2', 'col3': 'answ3'}
>>> data['col1']
'answ1'
>>> data[['col1']]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unhashable type: 'list'