API: disallow div/floordiv/pow operators for BooleanArray ?

[jorisvandenbossche]

Currently, for the plain bool dtype we explicitly check for some operations and raise an error, while those actually work in numpy. For example:

>>> arr = np.array([True, False, True])
>>> arr / True
array([1., 0., 1.])

>>> pd.Series(arr) / True
...
NotImplementedError: operator '/' not implemented for bool dtypes

This is done for the division and power operations (not_allowed={"/", "//", "**"}):

pandas/pandas/core/computation/expressions.py

Lines 215 to 218 in 934cad6

	if op_str in not_allowed:
	raise NotImplementedError(
	f"operator {repr(op_str)} not implemented for bool dtypes"
	)

For the nullable BooleanArray, for now we simply relied on the operations as defined by the underlying numpy bool array:

>>> pd.array(arr) / True
<FloatingArray>
[1.0, 0.0, 1.0]
Length: 3, dtype: Float64

That's for the BooleanArray, but the check is currently done on the "array_op" level (but because it is done within expressions.py, we don't run that check for EAs, xref #41161).

So questions:

• Do we actually want to continue disallowing this operation, while numpy allows it? (also, eg pd.Series(arr) / 1 does work, it's only disallowed if both operands are boolean)
• If we keep disallowing it, we should probably also disallow it on the BooleanArray level, and not only check it on the DataFrame/Series ops level in array_ops.py ?

[dsaxton]

@jorisvandenbossche What's the argument for disallowing at the moment? To me it seems more natural / Pythonic to allow this operation since booleans are essentially ints.

[jorisvandenbossche]

I am not sure what the historic reasons are.
Maybe because those operations were regarded as not that useful (although I would say it's up to the user to decide that), or as potentially confusing because they don't have a "boolean" interpretation, but only a numerical one (eg + and * still are a boolean operation resulting in booleans, and eg - raises an error in numpy about not being supported for booleans).
It's only for division and power that the booleans are interpreted as numeric values (and currently in pandas the user needs to be explicit about such casting).

[dsaxton]

@jorisvandenbossche Yeah, I can see the argument of "why would anyone do this?" but if it's easy to allow and makes for greater consistency with numpy and Python more generally, I personally like the idea of allowing this rather than making a special case here.

[jbrockmendel]

conditional on the Series behavior (which i wouldnt object to deprecating), I lean towards having BooleanArray behave like Series, i.e. raising here

[jbrockmendel]

Current thought here: BooleanArray (and IntegerArray and FloatingArray) ops should be wrappers around their core.ops.array_ops counterparts. This will allow for pushing more logic down from BooleanArray/NumericArray into BaseMaskedArray, which in turn will make it easier to extend BaseMaskedArray to wrap arbitrary dtypes.