Fix range index.

dmlc · Jul 2, 2021 · 6506bd6 · 6506bd6
1 parent c30c208
commit 6506bd6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python-package/xgboost/data.py b/python-package/xgboost/data.py
@@ -197,7 +197,7 @@ def _is_modin_df(data):
 def _transform_pandas_df(data, enable_categorical,
                          feature_names=None, feature_types=None,
                          meta=None, meta_type=None):
-    from pandas import MultiIndex, Int64Index
+    from pandas import MultiIndex, Int64Index, RangeIndex
     from pandas.api.types import is_sparse, is_categorical_dtype
 
     data_dtypes = data.dtypes
@@ -219,8 +219,8 @@ def _transform_pandas_df(data, enable_categorical,
             feature_names = [
                 ' '.join([str(x) for x in i]) for i in data.columns
             ]
-        elif isinstance(data.columns, Int64Index):
-            feature_names = list(map(lambda c: str(c).strip(), data.columns))
+        elif isinstance(data.columns, (Int64Index, RangeIndex)):
+            feature_names = list(map(str, data.columns))
         else:
             feature_names = data.columns.format()