BryanCutler

BryanCutler

BryanCutler

class SpanOpMixin:
  def __add__(self, other) -> Union["Span", "SpanArray"]:
 
    if isinstance(other, (ABCDataFrame, ABCSeries, ABCIndexClass)):
      # Rely on pandas to unbox and dispatch to us.
      return NotImplemented

    if isinstance(self, Span) and isinstance(other, Span):
      # Span + *Span = Span
      return Span(self.target_text,

BryanCutler

# Reduce the `SpanArray` to a single `Span` covering tokens 2 up to 5.
df["span"].iloc[2:5].sum()

# [6, 20): 'Python and the'

BryanCutler

@pytest.fixture
def dtype():
    """ Return dtype of your extension array."""
    return TensorDtype()
 
@pytest.fixture
def data(dtype):
    """ Return an extension array as data for the tests."""
    return pd.array(np.array([[i] for i in range(100)]), dtype=dtype)
  

BryanCutler

# Addition of `Span` with Series of `SpanDtype` produces another Series
df["span"].iloc[1]  + df["span"].iloc[3:5]

# 3    [0, 16): 'Monty Python and'
# 4    [0, 20): 'Monty Python and the'
# Name: span, dtype: SpanDtype