Función: extraer partición de red bipartita

src/redsem/main.py

@@ -7,3 +7,11 @@ def filtrar_primeras_posiciones(df, ultima=3):
     filtro = df['posicion'].isin(rango)
     df_filtrado = df[filtro].reset_index(drop=True)
     return df_filtrado
+
+
+def extraer_particion(red, particion):
+    assert nx.bipartite.is_bipartite(red), 'La red no es bipartita'
+    nodos_en_particion = [n for n, d in red.nodes(data=True) if d['bipartita'] == particion]
+    if len(nodos_en_particion) == 0:
+        raise ValueError(f'No hay nodos en la partición "{particion}"')
+    return nodos_en_particion