如何实现二维数组，以便我们可以轻松指定需要的列？

Question

mlb_data = pd.read_csv(file)

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

X = mlb_data[melbourne_features]

X现在将仅包含我们需要的列（在melbourne_features中指定）。如何实施？我尝试使用字典，二维数组和集合。但是似乎没有任何作用。

编辑：尝试使用字典：

>>> 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'