VectorDtype definition for Pandas Extension Types example

VectorDtype.py

import numpy as np
import pandas as pd
from pandas.core.dtypes.dtypes import PandasExtensionDtype
from pandas.api.extensions import ExtensionArray, ExtensionScalarOpsMixin, register_extension_dtype

@register_extension_dtype
class VectorDtype(PandasExtensionDtype):
    """
    Class to describe the custom Vector data type
    """
    type = Vector       # Scalar type for data
    name = 'vector'     # String identifying the data type name 

    @classmethod
    def construct_array_type(cls):
        """
        Return array type associated with this dtype
        """
        return VectorArray

    def __str__(self):
        return self.name