From 7821ad1eb295712fa552f71b76178ae7c5e73162 Mon Sep 17 00:00:00 2001
From: Yassin <aliyassin4@hotmail.com>
Date: Wed, 10 May 2023 02:12:19 +0200
Subject: [PATCH] Refactorings Code and Adding Node Filter
---
netbone/filters.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/netbone/filters.py b/netbone/filters.py
index 6b241cf..4ce8fef 100644
--- a/netbone/filters.py
+++ b/netbone/filters.py
@@ -45,7 +45,6 @@ def fraction_filter(backbone, value, narrate=True, secondary_property='weight',
filter_by = [backbone.property_name]
ascending = [backbone.ascending]
- value = math.ceil(value * len(data))
if backbone.filter_on == 'Edges':
filter_by.append(secondary_property)
@@ -58,9 +57,11 @@ def fraction_filter(backbone, value, narrate=True, secondary_property='weight',
backbone.narrate()
if backbone.filter_on == 'Edges':
+ value = math.ceil(value * len(data))
return nx.from_pandas_edgelist(data[:value], edge_attr=edge_properties(data))
else:
b = backbone.graph.copy()
+ value = math.ceil(value * len(backbone.graph))
b.remove_nodes_from(list(data[:value].index))
return b
--
GitLab