Импортируем необходимые модули, открываем PYMOL и загружаем структуру 1c11. Далее создаем список аминокислот белка, получаем изображение и создаем фильм для просмотра структуры.

In [2]:
import __main__
__main__.pymol_argv = [ 'pymol', '-x' ]

import pymol
pymol.finish_launching()
from pymol import cmd,stored

cmd.do('''
fetch 1cll, async=0
as lines, n. C+O+N+CA
zoom i. 4+5
mset 1 x1000
mview store''')
In [3]:
stored.r = [] 
cmd.iterate('1cll and n. CA','stored.r.append(int(resi))')
Out[3]:
152
In [13]:
import numpy as np

length = len(stored.r)
colors = np.linspace(1,0.5, length)
for k,i in enumerate(stored.r):
    cmd.set_color('col%d' %k, [colors[k],0.5,0.75])
    print [1,1,colors[k]]
    cmd.set('cartoon_color','col%d' % k ,'resi %d' % i)
cmd.show_as('cartoon','all')
In [14]:
cmd.zoom()
In [16]:
cmd.set("opaque_background", 'off')
cmd.set("ray_trace_mode", '4')
cmd.bg_color('white')
In [17]:
cmd.png('C:\Users\kseni\Desktop\mypng.png')
time.sleep(2)
Image(filename='C:\Users\kseni\Desktop\mypng.png')
Out[17]:
In [33]:
for i in range(length):
    cmd.frame((10*i)+1)
    cmd.zoom( 'n. CA and i. %d+%d' % (i,i+7))
    cmd.mview('store')
cmd.mplay()    
#видео проигрывается в окне PYMOL
In [ ]: