radugrosu

import numpy as np
from typing import Sequence

class NamedArray(np.ndarray):
    def __new__(cls, array: np.ndarray, names: Sequence[str]):
        assert array.shape[-1] == len(names)
        obj = np.asarray(array).view(cls)
        obj.names = names
        obj._lookup = dict(zip(names, range(len(array))))
        return obj