def get_col_range(self, group_by: tp.GroupByLike = None) -> tp.ColRange:
"""Get group-aware column range."""
if not self.wrapper.grouper.is_grouped(group_by=group_by):
return self.col_range
col_arr = self.get_col_arr(group_by=group_by)
columns = self.wrapper.get_columns(group_by=group_by)
return nb.col_range_nb(col_arr, len(columns))
def col_range(self) -> tp.ColRange:
"""Column index.
Faster than `ColumnMapper.col_map` but only compatible with sorted columns.
More suited for records."""
return nb.col_range_nb(self.col_arr, len(self.wrapper.columns))