diff --git a/netbone/filters.py b/netbone/filters.py index 6b241cf23dafdba6371b5cc27432b4e3109d45ad..4ce8fef0a0c847c35a3a8c42fe0dca90f7077615 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