ENH: nullable Float32/64 ExtensionArray

Author: jorisvandenbossche

pandas/__init__.py

@@ -59,6 +59,8 @@
     UInt16Dtype,
     UInt32Dtype,
     UInt64Dtype,
+    Float32Dtype,
+    Float64Dtype,
     CategoricalDtype,
     PeriodDtype,
     IntervalDtype,

pandas/arrays/__init__.py

@@ -7,6 +7,7 @@
     BooleanArray,
     Categorical,
     DatetimeArray,
+    FloatingArray,
     IntegerArray,
     IntervalArray,
     PandasArray,
@@ -20,6 +21,7 @@
     "BooleanArray",
     "Categorical",
     "DatetimeArray",
+    "FloatingArray",
     "IntegerArray",
     "IntervalArray",
     "PandasArray",

pandas/core/api.py

@@ -14,6 +14,7 @@
 from pandas.core.algorithms import factorize, unique, value_counts
 from pandas.core.arrays import Categorical
 from pandas.core.arrays.boolean import BooleanDtype
+from pandas.core.arrays.floating import Float32Dtype, Float64Dtype
 from pandas.core.arrays.integer import (
     Int8Dtype,
     Int16Dtype,

pandas/core/arrays/__init__.py

@@ -6,6 +6,7 @@
 from pandas.core.arrays.boolean import BooleanArray
 from pandas.core.arrays.categorical import Categorical
 from pandas.core.arrays.datetimes import DatetimeArray
+from pandas.core.arrays.floating import FloatingArray
 from pandas.core.arrays.integer import IntegerArray, integer_array
 from pandas.core.arrays.interval import IntervalArray
 from pandas.core.arrays.numpy_ import PandasArray, PandasDtype
@@ -21,6 +22,7 @@
     "BooleanArray",
     "Categorical",
     "DatetimeArray",
+    "FloatingArray",
     "IntegerArray",
     "integer_array",
     "IntervalArray",

pandas/core/arrays/boolean.py

@@ -604,10 +604,11 @@ def logical_method(self, other):
     @classmethod
     def _create_comparison_method(cls, op):
         def cmp_method(self, other):
-            from pandas.arrays import IntegerArray
+            from pandas.arrays import IntegerArray, FloatingArray
 
             if isinstance(
-                other, (ABCDataFrame, ABCSeries, ABCIndexClass, IntegerArray)
+                other,
+                (ABCDataFrame, ABCSeries, ABCIndexClass, IntegerArray, FloatingArray),
             ):
                 # Rely on pandas to unbox and dispatch to us.
                 return NotImplemented