Colour coding modules in python from other text

[kirk86]

I was wondering if zenburn can colour code modules imported and used in python code in order to separate them from other texts.

For instance in the snippet below

I'd like whenever I use np. prefix of numpy module to be highlighted in some distinct colour in order to easily separate them from the remaining text, similar with other modules like pd. for pandas.

Essentially any module that is somehow imported in python code should be easily distinct and visible from the rest of the text code, that's why using some colour for standard modules like numpy, pandas, etc, and another colour for user defined modules would really help along a lot when coding, especially in python.