Source code for static_frame.core.fill_value_auto


import typing as tp

import numpy as np

from static_frame.core.util import NAT
from static_frame.core.util import NAT_TD64

FILL_VALUE_UNSET = object()

[docs]class FillValueAuto: '''Define, per NumPy dtype kind, a value to be used for filling missing values. ''' __slots__ = tuple('biufcmMOSUV')
[docs] @classmethod def from_default(cls, b: tp.Any = False, # np.bool_(False) i: tp.Any = 0, u: tp.Any = 0, f: tp.Any = np.nan, c: tp.Any = complex(np.nan, np.nan), m: tp.Any = NAT_TD64, M: tp.Any = NAT, O: tp.Any = None, S: tp.Any = b'', U: tp.Any = '', V: tp.Any = b'\0', ) -> 'FillValueAuto': '''Create a ``FileValueAuto`` instance based on a default selected to prohibit type coercions. Args: b: fill value for bool kind i: fill value for integer kind u: fill value for unsigned integer kind f: fill value for float kind c: fill value for complex kind m: fill value for timedelta62 kind M: fill value for datetime64 kind O: fill value for object kind S: fill value for bytes kind U: fill value for unicode kind V: fill value for void kind ''' return cls( b=b, i=i, u=u, f=f, c=c, m=m, M=M, O=O, S=S, U=U, V=V, )
[docs] def __init__(self, b: tp.Any = FILL_VALUE_UNSET, i: tp.Any = FILL_VALUE_UNSET, u: tp.Any = FILL_VALUE_UNSET, f: tp.Any = FILL_VALUE_UNSET, c: tp.Any = FILL_VALUE_UNSET, m: tp.Any = FILL_VALUE_UNSET, M: tp.Any = FILL_VALUE_UNSET, O: tp.Any = FILL_VALUE_UNSET, S: tp.Any = FILL_VALUE_UNSET, U: tp.Any = FILL_VALUE_UNSET, V: tp.Any = FILL_VALUE_UNSET, ) -> None: ''' Args: b: fill value for bool kind i: fill value for integer kind u: fill value for unsigned integer kind f: fill value for float kind c: fill value for complex kind m: fill value for timedelta62 kind M: fill value for datetime64 kind O: fill value for object kind S: fill value for bytes kind U: fill value for unicode kind V: fill value for void kind ''' self.b = b self.i = i self.u = u self.f = f self.c = c self.m = m self.M = M self.O = O self.S = S self.U = U self.V = V
[docs] def __getitem__(self, dtype: np.dtype) -> tp.Any: fv = getattr(self, dtype.kind) if fv is FILL_VALUE_UNSET: raise RuntimeError(f'FillValueAuto requested value for kind {dtype.kind} not defined.') return fv