def from_pandas(cls, value, *, is_go: bool = False) -> 'IndexBase':
'''
Given a Pandas index, return the appropriate IndexBase derived class.
'''
import pandas
from static_frame import Index
from static_frame import IndexGO
from static_frame import IndexDate
from static_frame import IndexHierarchy
from static_frame import IndexHierarchyGO
if isinstance(value, pandas.MultiIndex):
# iterating over a hierarchucal index will iterate over labels
if is_go:
return IndexHierarchyGO.from_labels(value)
return IndexHierarchy.from_labels(value)
elif isinstance(value, pandas.DatetimeIndex):
if is_go:
raise NotImplementedError(
'No grow-only version of IndexDate yet exists')
return IndexDate(value)
if is_go:
return IndexGO(value)
return Index(value)
def from_pandas(
cls,
value: 'pandas.DataFrame',
) -> 'IndexBase':
'''
Given a Pandas index, return the appropriate IndexBase derived class.
'''
import pandas
from static_frame import Index
from static_frame import IndexGO
from static_frame import IndexHierarchy
from static_frame import IndexHierarchyGO
from static_frame import IndexNanosecond
from static_frame import IndexNanosecondGO
from static_frame.core.index_datetime import IndexDatetime
if isinstance(value, pandas.MultiIndex):
# iterating over a hierarchucal index will iterate over labels
name = tuple(value.names)
if not cls.STATIC:
return IndexHierarchyGO.from_labels(value, name=name)
return IndexHierarchy.from_labels(value, name=name)
elif isinstance(value, pandas.DatetimeIndex):
# if IndexDatetime, use cls, else use IndexNanosecond
if issubclass(cls, IndexDatetime):
return cls(value, name=value.name)
else:
if not cls.STATIC:
return IndexNanosecondGO(value, name=value.name)
return IndexNanosecond(value, name=value.name)
if not cls.STATIC:
return IndexGO(value, name=value.name)
return Index(value, name=value.name)
def from_pandas(cls,
value: 'pandas.DataFrame',
) -> 'IndexBase':
'''
Given a Pandas index, return the appropriate IndexBase derived class.
'''
import pandas
from static_frame.core.index_datetime import IndexDatetime
from static_frame import Index
from static_frame import IndexGO
from static_frame import IndexHierarchy
from static_frame import IndexHierarchyGO
if isinstance(value, pandas.MultiIndex):
# iterating over a hierarchucal index will iterate over labels
name = tuple(value.names)
if not cls.STATIC:
return IndexHierarchyGO.from_labels(value, name=name)
return IndexHierarchy.from_labels(value, name=name)
elif isinstance(value, pandas.DatetimeIndex):
# coming from a Pandas datetime index, in the absence of other information, the best match is a Nanosecond index
if not issubclass(cls, IndexDatetime):
raise ErrorInitIndex(f'cannot create a datetime Index from {cls}')
if not cls.STATIC:
return cls(value, name=value.name)
return cls(value, name=value.name)
if not cls.STATIC:
return IndexGO(value, name=value.name)
return Index(value, name=value.name)
def test_index_from_optional_constructor_c(self) -> None:
idx0 = IndexHierarchyGO.from_labels([('a', 0), ('a', 1), ('b', 0),
('b', 1)])
idx1 = index_from_optional_constructor(
idx0, default_constructor=IndexHierarchy.from_labels)
# Since the default constructo is static, we should be able to reuse the index
self.assertNotEqual(id(idx0), id(idx1))
self.assertTrue(idx1.STATIC)
def from_pandas(
cls,
value: 'pandas.Index',
) -> 'IndexBase':
'''
Given a Pandas index, return the appropriate IndexBase derived class.
'''
import pandas
if not isinstance(value, pandas.Index):
raise ErrorInitIndex(
f'from_pandas must be called with a Pandas Index object, not: {type(value)}'
)
from static_frame import Index
from static_frame import IndexGO
from static_frame import IndexHierarchy
from static_frame import IndexHierarchyGO
from static_frame import IndexNanosecond
from static_frame import IndexNanosecondGO
from static_frame.core.index_datetime import IndexDatetime
if isinstance(value, pandas.MultiIndex):
# iterating over a hierarchical index will iterate over labels
name: tp.Optional[tp.Tuple[tp.Hashable, ...]] = tuple(value.names)
# if not assigned Pandas returns None for all components, which will raise issue if trying to unset this index.
if all(n is None for n in name): #type: ignore
name = None
depth = value.nlevels
if not cls.STATIC:
return IndexHierarchyGO.from_labels(value,
name=name,
depth_reference=depth)
return IndexHierarchy.from_labels(value,
name=name,
depth_reference=depth)
elif isinstance(value, pandas.DatetimeIndex):
# if IndexDatetime, use cls, else use IndexNanosecond
if issubclass(cls, IndexDatetime):
return cls(value, name=value.name)
else:
if not cls.STATIC:
return IndexNanosecondGO(value, name=value.name)
return IndexNanosecond(value, name=value.name)
if not cls.STATIC:
return IndexGO(value, name=value.name)
return Index(value, name=value.name)
def test_hierarchy_rehierarch_b(self) -> None:
labels = (
('I', 'A'),
('I', 'B'),
('II', 'C'),
('II', 'B'),
('II', 'D'),
('III', 'D'),
('IV', 'A'),
)
ih1 = IndexHierarchyGO.from_labels(labels)
self.assertEqual(
ih1.rehierarch([1, 0]).values.tolist(),
[['A', 'I'], ['A', 'IV'], ['B', 'I'], ['B', 'II'], ['C', 'II'],
['D', 'II'], ['D', 'III']])
def test_hierarchy_copy_b(self):
labels = (
('I', 'A'),
('I', 'B'),
('II', 'A'),
('II', 'B'),
)
ih1 = IndexHierarchyGO.from_labels(labels)
ih2 = ih1.copy()
ih2.append(('II', 'C'))
self.assertEqual(
ih2.values.tolist(),
[['I', 'A'], ['I', 'B'], ['II', 'A'], ['II', 'B'], ['II', 'C']])
self.assertEqual(ih1.values.tolist(),
[['I', 'A'], ['I', 'B'], ['II', 'A'], ['II', 'B']])