Skip to content
Snippets Groups Projects
Commit 6398c09f authored by dmt's avatar dmt
Browse files

Provide capability to find overlapping rows and fuse learnblocks.

parent a8a8f3fe
No related branches found
No related tags found
No related merge requests found
...@@ -45,6 +45,7 @@ class PandasBlock: ...@@ -45,6 +45,7 @@ class PandasBlock:
self.__data_block = data_block self.__data_block = data_block
self.relatives = relatives self.relatives = relatives
self.n_cluster = None self.n_cluster = None
self.purpose = None
def __str__(self): def __str__(self):
return str(self.__data_block) return str(self.__data_block)
...@@ -82,6 +83,17 @@ class PandasBlock: ...@@ -82,6 +83,17 @@ class PandasBlock:
data_frame["Z"] = labels data_frame["Z"] = labels
return PandasBlock(data_frame, self.relatives) return PandasBlock(data_frame, self.relatives)
def overlapping_rows(self, block):
overlapping = self.__data_block[self.__data_block.isin(block)].dropna(
axis=0, how="all"
)
return self.new_block_from_rows_index(overlapping.indexes)
def fusion(self, block):
return self.new_block_from_rows_index(
self.indexes+block.indexes
)
@property @property
def min_timestamp(self): def min_timestamp(self):
return min(self.__data_block["T"]) return min(self.__data_block["T"])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment