Función: filtrar red por grupo de estímulos junto con sus respuestas asociadas

src/redsem/main.py

@@ -24,3 +24,10 @@ def obtener_respuestas_asociadas(red, *estimulos):
     for estimulo in estimulos:
     for estimulo in estimulos:
         respuestas_asociadas += list(red.neighbors(estimulo))
         respuestas_asociadas += list(red.neighbors(estimulo))
     return respuestas_asociadas
     return respuestas_asociadas
+
+
+def filtrar_red_por_estimulos(red, *estimulos):
+    respuestas_asociadas = obtener_respuestas_asociadas(red, *estimulos)
+    estimulos_y_respuestas = respuestas_asociadas + list(estimulos)
+    red_filtrada = red.subgraph(estimulos_y_respuestas).copy()
+    return red_filtrada