Define helper methods needed in construction and feature selection.

cml/ports/source_adapters.py

@@ -43,20 +43,44 @@ class PandasBlock:
     def __init__(self, data_block, relatives=None):
         self.__data_block = data_block
         self.relatives = relatives
+        self.n_cluster = None
 
     def __str__(self):
         return str(self.__data_block)
 
+    def __repr__(self):
+        return str(
+            self.__data_block[
+                self.__data_block.columns[:self.rows-self._LAST_THREE_COLUMNS]])
+
     def __getitem__(self, item):
         return self.__data_block.iloc[item][:self.rows-self._LAST_THREE_COLUMNS]
 
     def __len__(self):
         return self.__data_block.shape[0]
 
+    def as_numpy_array(self):
+        return self.__data_block[
+            self.__data_block.columns[
+                :self.rows - self._LAST_THREE_COLUMNS]].values
+
     def set_labels(self, labels):
-        data_frame = self.__data_block["Z"] = labels
+        data_frame = self.__data_block.copy()
+        data_frame["Z"] = labels
         return PandasBlock(data_frame, self.relatives)
 
+    @property
+    def min_timestamp(self):
+        return min(self.__data_block["T"])
+
+    @property
+    def max_timestamp(self):
+        return max(self.__data_block["T"])
+
+    @property
+    def learn_rows(self):
+        return self.__data_block.shape[1] - 3
+
     @property
     def rows(self):
         return self.__data_block.shape[1]
@@ -110,6 +134,22 @@ class PandasBlock:
     def get_column_values(self, column_name):
         return self.__data_block[column_name]
 
+    def get_column_name_by_index(self, index):
+        column_names = self.__data_block.column
+        return column_names[index]
+
+    @property
+    def feature_count(self):
+        return self.__data_block.shape[1] - self._LAST_THREE_COLUMNS
+
+    def drop_columns_by_index(self, index):
+        if isinstance(index, set):
+            remove_columns = [self.__data_block.columns[i] for i in index]
+        else:
+            remove_columns = index
+
+        self.__data_block.drop(remove_columns, axis=1, inplace=True)
+
 
 class PandasAdapter:
     def __init__(self, data_frame):