<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def get_values(self, idx):
 """ Return the variable values at the given indices """

        if isinstance(idx, list):
            idx = np.array(idx, dtype=int)

        return self._data[:, idx]