compensate function should return all columns from input data

[whitews]

Currently, the compensate function takes a numpy data array as an input along with the list of column indices that will be compensated. However, it only returns the compensated columns, requiring the caller to re-assemble the original data array with the compensated columns. This is a bit awkward and the function should return the whole data array.