Partition electron density and visualize cube file


I have a question that I could use this post (ESP Surfaces with Psi4) as an example.

Is there a way that I could partition the electron density to their respective atom in a water molecule in order to visualize them in different colors on VMD?

Thank you,