diff --git a/cml/ports/source_adapters.py b/cml/ports/source_adapters.py index 19222420bdce811c9d08d4d39c6339c671e1c3ad..08fc976b87fa226d231a62f7c81b13f981e2d85d 100644 --- a/cml/ports/source_adapters.py +++ b/cml/ports/source_adapters.py @@ -98,6 +98,10 @@ class PandasBlock: def rows(self): return self.__data_block.shape[1] + def new_block_from_rows_index(self, indices: List[int]): + data_from = self.__data_block.loc[indices] + return PandasBlock(data_from) + def new_block_from(self, column_values): data_from = self.__data_block.loc[self.__data_block["T"].isin( column_values)]