def mgr_to_mgr(mgr, typ: str, copy: bool = True):
"""
Convert to specific type of Manager. Does not copy if the type is already
correct. Does not guarantee a copy otherwise. `copy` keyword only controls
whether conversion from Block->ArrayManager copies the 1D arrays.
"""
new_mgr: Manager
if typ == "block":
if isinstance(mgr, BlockManager):
new_mgr = mgr
else:
if mgr.ndim == 2:
new_mgr = arrays_to_mgr(
mgr.arrays, mgr.axes[0], mgr.axes[1], typ="block"
)
else:
new_mgr = SingleBlockManager.from_array(mgr.arrays[0], mgr.index)
elif typ == "array":
if isinstance(mgr, ArrayManager):
new_mgr = mgr
else:
if mgr.ndim == 2:
arrays = [mgr.iget_values(i) for i in range(len(mgr.axes[0]))]
if copy:
arrays = [arr.copy() for arr in arrays]
new_mgr = ArrayManager(arrays, [mgr.axes[1], mgr.axes[0]])
else:
array = mgr.internal_values()
if copy:
array = array.copy()
new_mgr = SingleArrayManager([array], [mgr.index])
else:
raise ValueError(f"'typ' needs to be one of {{'block', 'array'}}, got '{typ}'")
return new_mgr
def pandas_series_resolver(obj, resolver):
meta = obj.meta
name = json.loads(meta['name'])
index = resolver.run(obj.member('index_'))
np_value = resolver.run(obj.member('value_'))
block = Block(np_value, slice(0, len(np_value), 1), ndim=1)
return pd.Series(SingleBlockManager(block, index), name=name)
def mgr_to_mgr(mgr, typ: str):
"""
Convert to specific type of Manager. Does not copy if the type is already
correct. Does not guarantee a copy otherwise.
"""
new_mgr: Manager
if typ == "block":
if isinstance(mgr, BlockManager):
new_mgr = mgr
else:
if mgr.ndim == 2:
new_mgr = arrays_to_mgr(
mgr.arrays, mgr.axes[0], mgr.axes[1], mgr.axes[0], typ="block"
)
else:
new_mgr = SingleBlockManager.from_array(mgr.arrays[0], mgr.index)
elif typ == "array":
if isinstance(mgr, ArrayManager):
new_mgr = mgr
else:
if mgr.ndim == 2:
arrays = [mgr.iget_values(i).copy() for i in range(len(mgr.axes[0]))]
new_mgr = ArrayManager(arrays, [mgr.axes[1], mgr.axes[0]])
else:
new_mgr = SingleArrayManager([mgr.internal_values()], [mgr.index])
else:
raise ValueError(f"'typ' needs to be one of {{'block', 'array'}}, got '{typ}'")
return new_mgr
def iget(self, i: int) -> SingleBlockManager:
"""
Return the data as a SingleBlockManager.
"""
from pandas.core.internals.managers import SingleBlockManager
values = self.arrays[i]
block = make_block(values, placement=slice(0, len(values)), ndim=1)
return SingleBlockManager(block, self._axes[0])
def pandas_series_resolver(obj, resolver):
meta = obj.meta
name = from_json(meta['name'])
index = resolver.run(obj.member('index_'))
np_value = resolver.run(obj.member('value_'))
if BlockPlacement:
placement = BlockPlacement(slice(0, len(np_value), 1))
else:
placement = slice(0, len(np_value), 1)
block = Block(np_value, placement, ndim=1)
return pd.Series(SingleBlockManager(block, index), name=name)
