diff --git a/cml/ports/source_adapters.py b/cml/ports/source_adapters.py index 2415b694e2c1da6dfc9b339b57260e153664a073..537491fdf9c35ed858e823ca3b83ab82842e9564 100644 --- a/cml/ports/source_adapters.py +++ b/cml/ports/source_adapters.py @@ -99,7 +99,13 @@ class PandasBlock: @property def length(self): - return len(self.__data_block) + return self.__data_block.shape[0] + + def drop_row(self, index): + self.__data_block.drop(index, inplace=True) + + def get_column_values(self, column_name): + return self.__data_block[column_name] class PandasAdapter: @@ -113,7 +119,10 @@ class PandasAdapter: @property def length(self): - return len(self.__data_frame) + return self.__data_frame.shape[0] + + def get_block_via_index(self, indexes): + return PandasBlock(self.__data_frame.iloc[indexes]) def get_block(self, start, end=None, step=None): return PandasBlock(self.__data_frame[start:end:step])