Isso pode não ser mais útil para o autor, mas eu coloquei o seguinte em .vim/ftplugin/python.vim
:
if exists('+colorcolumn')
setlocal colorcolumn=81
else
au! BufEnter <buffer> match ColorColumn /\%81v.*/
endif
Como está no ftplugin, ele ocorre apenas para arquivos python, e o BufEnter o mantém local no buffer em que o arquivo python está.