add deprecation trigger in pd.__init__.py

Author: topper-123

pandas/__init__.py

@@ -71,7 +71,6 @@
     # indexes
     Index,
     CategoricalIndex,
-    Int64Index,
     UInt64Index,
     RangeIndex,
     Float64Index,
@@ -187,10 +186,41 @@
 
 
 # GH 27101
+deprecated_num_index_names = ["Float64Index", "Int64Index", "UInt64Index"]
+
+
+def __dir__():
+    return list(globals().keys()) + deprecated_num_index_names
+
+
 def __getattr__(name):
     import warnings
 
-    if name == "datetime":
+    if name in deprecated_num_index_names:
+        num_msg = (
+            f"The pandas.{name} class is deprecated "
+            "and will be removed from pandas in a future version. "
+            "Use pandas.NumericIndex with the appropriate dtype instead."
+        )
+        if name == "Int64Index":
+            from pandas.core.api import Int64Index
+
+            warnings.warn(num_msg, FutureWarning, stacklevel=2)
+            return Int64Index
+        elif name == "Unt64Index":
+            from pandas.core.api import UInt64Index
+
+            warnings.warn(num_msg, FutureWarning, stacklevel=2)
+            return UInt64Index
+        elif name == "Float64Index":
+            from pandas.core.api import Float64Index
+
+            warnings.warn(num_msg, FutureWarning, stacklevel=2)
+            return Float64Index
+        else:
+            raise NameError(name)
+
+    elif name == "datetime":
         warnings.warn(
             "The pandas.datetime class is deprecated "
             "and will be removed from pandas in a future version. "

pandas/_testing/__init__.py

@@ -45,15 +45,12 @@
     CategoricalIndex,
     DataFrame,
     DatetimeIndex,
-    Float64Index,
     Index,
-    Int64Index,
     IntervalIndex,
     MultiIndex,
     NumericIndex,
     RangeIndex,
     Series,
-    UInt64Index,
     bdate_range,
 )
 from pandas._testing._io import (  # noqa:F401
@@ -106,6 +103,11 @@
     use_numexpr,
     with_csv_dialect,
 )
+from pandas.core.api import (
+    Float64Index,
+    Int64Index,
+    UInt64Index,
+)
 from pandas.core.arrays import (
     DatetimeArray,
     PandasArray,

pandas/core/algorithms.py

@@ -1334,7 +1334,7 @@ def __init__(self, obj, n: int, keep: str, columns):
 
     def compute(self, method: str) -> DataFrame:
 
-        from pandas import Int64Index
+        from pandas.core.api import Int64Index
 
         n = self.n
         frame = self.obj

pandas/core/indexes/base.py

@@ -572,15 +572,15 @@ def _dtype_to_subclass(cls, dtype: DtypeObj):
             return TimedeltaIndex
 
         elif is_float_dtype(dtype):
-            from pandas import Float64Index
+            from pandas.core.api import Float64Index
 
             return Float64Index
         elif is_unsigned_integer_dtype(dtype):
-            from pandas import UInt64Index
+            from pandas.core.api import UInt64Index
 
             return UInt64Index
         elif is_signed_integer_dtype(dtype):
-            from pandas import Int64Index
+            from pandas.core.api import Int64Index
 
             return Int64Index
 

pandas/io/feather_format.py

@@ -9,12 +9,12 @@
 from pandas.compat._optional import import_optional_dependency
 from pandas.util._decorators import doc
 
-from pandas import (
+from pandas.core import generic
+from pandas.core.api import (
     DataFrame,
     Int64Index,
     RangeIndex,
 )
-from pandas.core import generic
 
 from pandas.io.common import get_handle
 

pandas/io/pytables.py

@@ -65,14 +65,14 @@
     DataFrame,
     DatetimeIndex,
     Index,
-    Int64Index,
     MultiIndex,
     PeriodIndex,
     Series,
     TimedeltaIndex,
     concat,
     isna,
 )
+from pandas.core.api import Int64Index
 from pandas.core.arrays import (
     Categorical,
     DatetimeArray,

pandas/tests/frame/constructors/test_from_records.py

@@ -11,7 +11,6 @@
     CategoricalIndex,
     DataFrame,
     Index,
-    Int64Index,
     Interval,
     RangeIndex,
     Series,
@@ -450,7 +449,7 @@ def test_from_records_empty_with_nonempty_fields_gh3682(self):
         a = np.array([(1, 2)], dtype=[("id", np.int64), ("value", np.int64)])
         df = DataFrame.from_records(a, index="id")
 
-        ex_index = Int64Index([1], name="id")
+        ex_index = Index([1], name="id")
         expected = DataFrame({"value": [2]}, index=ex_index, columns=["value"])
         tm.assert_frame_equal(df, expected)
 

pandas/tests/frame/methods/test_truncate.py

@@ -4,10 +4,12 @@
 import pandas as pd
 from pandas import (
     DataFrame,
+    DatetimeIndex,
     Series,
     date_range,
 )
 import pandas._testing as tm
+from pandas.core.api import Int64Index
 
 
 class TestDataFrameTruncate:
@@ -116,13 +118,13 @@ def test_truncate_nonsortedindex_axis1(self):
         "before, after, indices",
         [(1, 2, [2, 1]), (None, 2, [2, 1, 0]), (1, None, [3, 2, 1])],
     )
-    @pytest.mark.parametrize("klass", [pd.Int64Index, pd.DatetimeIndex])
+    @pytest.mark.parametrize("klass", [Int64Index, DatetimeIndex])
     def test_truncate_decreasing_index(
         self, before, after, indices, klass, frame_or_series
     ):
         # https://github.com/pandas-dev/pandas/issues/33756
         idx = klass([3, 2, 1, 0])
-        if klass is pd.DatetimeIndex:
+        if klass is DatetimeIndex:
             before = pd.Timestamp(before) if before is not None else None
             after = pd.Timestamp(after) if after is not None else None
             indices = [pd.Timestamp(i) for i in indices]

pandas/tests/frame/test_constructors.py

@@ -53,6 +53,7 @@
     PeriodArray,
     SparseArray,
 )
+from pandas.core.api import Int64Index
 
 MIXED_FLOAT_DTYPES = ["float16", "float32", "float64"]
 MIXED_INT_DTYPES = [
@@ -580,7 +581,7 @@ def test_constructor_2d_index(self):
         df = DataFrame([[1]], columns=[[1]], index=[1, 2])
         expected = DataFrame(
             [1, 1],
-            index=pd.Int64Index([1, 2], dtype="int64"),
+            index=Int64Index([1, 2], dtype="int64"),
             columns=MultiIndex(levels=[[1]], codes=[[0]]),
         )
         tm.assert_frame_equal(df, expected)

pandas/tests/indexes/common.py

@@ -21,9 +21,7 @@
 from pandas import (
     CategoricalIndex,
     DatetimeIndex,
-    Float64Index,
     Index,
-    Int64Index,
     IntervalIndex,
     MultiIndex,
     NumericIndex,
@@ -33,8 +31,12 @@
     TimedeltaIndex,
     isna,
 )
-from pandas import UInt64Index  # noqa:F401
 import pandas._testing as tm
+from pandas.core.api import (  # noqa:F401
+    Float64Index,
+    Int64Index,
+    UInt64Index,
+)
 from pandas.core.indexes.datetimelike import DatetimeIndexOpsMixin
 
 

pandas/tests/indexes/datetimes/methods/test_astype.py

@@ -9,13 +9,13 @@
 from pandas import (
     DatetimeIndex,
     Index,
-    Int64Index,
     NaT,
     PeriodIndex,
     Timestamp,
     date_range,
 )
 import pandas._testing as tm
+from pandas.core.api import Int64Index
 
 
 class TestDatetimeIndex:

pandas/tests/indexes/datetimes/test_setops.py

@@ -10,12 +10,12 @@
     DataFrame,
     DatetimeIndex,
     Index,
-    Int64Index,
     Series,
     bdate_range,
     date_range,
 )
 import pandas._testing as tm
+from pandas.core.api import Int64Index
 
 from pandas.tseries.offsets import (
     BMonthEnd,

pandas/tests/indexes/interval/test_constructors.py

@@ -9,9 +9,7 @@
 from pandas import (
     Categorical,
     CategoricalIndex,
-    Float64Index,
     Index,
-    Int64Index,
     Interval,
     IntervalIndex,
     date_range,
@@ -20,6 +18,10 @@
     timedelta_range,
 )
 import pandas._testing as tm
+from pandas.core.api import (
+    Float64Index,
+    Int64Index,
+)
 from pandas.core.arrays import IntervalArray
 import pandas.core.common as com
 

pandas/tests/indexes/multi/test_drop.py

@@ -189,5 +189,5 @@ def test_droplevel_multiindex_one_level():
     # GH#37208
     index = MultiIndex.from_tuples([(2,)], names=("b",))
     result = index.droplevel([])
-    expected = pd.Int64Index([2], name="b")
+    expected = Index([2], name="b")
     tm.assert_index_equal(result, expected)

pandas/tests/indexes/multi/test_integrity.py

@@ -12,6 +12,7 @@
     RangeIndex,
 )
 import pandas._testing as tm
+from pandas.core.api import Int64Index
 
 
 def test_labels_dtypes():
@@ -83,17 +84,17 @@ def test_values_multiindex_periodindex():
     idx = MultiIndex.from_arrays([ints, pidx])
     result = idx.values
 
-    outer = pd.Int64Index([x[0] for x in result])
-    tm.assert_index_equal(outer, pd.Int64Index(ints))
+    outer = Int64Index([x[0] for x in result])
+    tm.assert_index_equal(outer, Int64Index(ints))
 
     inner = pd.PeriodIndex([x[1] for x in result])
     tm.assert_index_equal(inner, pidx)
 
     # n_lev > n_lab
     result = idx[:2].values
 
-    outer = pd.Int64Index([x[0] for x in result])
-    tm.assert_index_equal(outer, pd.Int64Index(ints[:2]))
+    outer = Int64Index([x[0] for x in result])
+    tm.assert_index_equal(outer, Int64Index(ints[:2]))
 
     inner = pd.PeriodIndex([x[1] for x in result])
     tm.assert_index_equal(inner, pidx[:2])
@@ -246,11 +247,11 @@ def test_rangeindex_fallback_coercion_bug():
     tm.assert_frame_equal(df, expected, check_like=True)
 
     result = df.index.get_level_values("fizz")
-    expected = pd.Int64Index(np.arange(10), name="fizz").repeat(10)
+    expected = Int64Index(np.arange(10), name="fizz").repeat(10)
     tm.assert_index_equal(result, expected)
 
     result = df.index.get_level_values("buzz")
-    expected = pd.Int64Index(np.tile(np.arange(10), 10), name="buzz")
+    expected = Int64Index(np.tile(np.arange(10), 10), name="buzz")
     tm.assert_index_equal(result, expected)
 
 

pandas/tests/indexes/multi/test_names.py

@@ -148,7 +148,7 @@ def test_setting_names_from_levels_raises():
         new.index.name = "bar"
 
     assert pd.Index._no_setting_name is False
-    assert pd.Int64Index._no_setting_name is False
+    assert pd.NumericIndex._no_setting_name is False
     assert pd.RangeIndex._no_setting_name is False
 
 

pandas/tests/indexes/period/test_indexing.py

@@ -23,6 +23,7 @@
     period_range,
 )
 import pandas._testing as tm
+from pandas.core.api import Int64Index
 
 dti4 = date_range("2016-01-01", periods=4)
 dti = dti4[:-1]
@@ -929,7 +930,7 @@ def test_asof_locs_mismatched_type(self):
 
         msg = "must be DatetimeIndex or PeriodIndex"
         with pytest.raises(TypeError, match=msg):
-            pi.asof_locs(pd.Int64Index(pi.asi8), mask)
+            pi.asof_locs(Int64Index(pi.asi8), mask)
 
         with pytest.raises(TypeError, match=msg):
             pi.asof_locs(pd.Float64Index(pi.asi8), mask)

pandas/tests/indexes/ranges/test_indexing.py

@@ -1,9 +1,9 @@
 import numpy as np
 import pytest
 
-import pandas as pd
 from pandas import RangeIndex
 import pandas._testing as tm
+from pandas.core.api import Int64Index
 
 
 class TestGetIndexer:
@@ -55,7 +55,7 @@ def test_take_fill_value(self):
         # GH#12631
         idx = RangeIndex(1, 4, name="xxx")
         result = idx.take(np.array([1, 0, -1]))
-        expected = pd.Int64Index([2, 1, 3], name="xxx")
+        expected = Int64Index([2, 1, 3], name="xxx")
         tm.assert_index_equal(result, expected)
 
         # fill_value
@@ -65,7 +65,7 @@ def test_take_fill_value(self):
 
         # allow_fill=False
         result = idx.take(np.array([1, 0, -1]), allow_fill=False, fill_value=True)
-        expected = pd.Int64Index([2, 1, 3], name="xxx")
+        expected = Int64Index([2, 1, 3], name="xxx")
         tm.assert_index_equal(result, expected)
 
         msg = "Unable to fill values because RangeIndex cannot contain NA"