Dynamically draw a graph with node changing color

Question

I want to draw the evolution of a graph where some nodes change color.

%matplotlib tk
import networkx as nx 
import matplotlib.pyplot as plt
import numpy as np
im