I prefer having different colorscheme for python and different for rest of the files.
colorscheme
So, what I did is added the below two lines in my vimrc
vimrc
It's not exactly what you want, but this tab page color scheme vim plugin lets you assign a different colorscheme for each tab with :Tcolorscheme.
:Tcolorscheme