ENH: Added get_columns_axis