1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
| def generate(input_str): ''' input_str = '[4,8]' return [4,5,6,7,8] ''' input_list = ast.literal_eval(input_str) res = [] for sublist in input_list: res += [i for i in range(sublist[0], sublist[1]+1)] return res
def show(num): exm_p = name[num] exm_i = index[num] exm_p_dir = os.path.join(pdb_dir, exm_p+'.pdb') save_dir = '/data/lwb/data/HumanProteins/View'
pc.load(exm_p_dir) pc.hide('everything') pc.color('gray') pc.bg_color('white')
pc.show('cartoon') pc.set('cartoon_transparency', 0.5)
sel = '+'.join([str(i) for i in generate(exm_i)]) print(sel) pc.select('cluster1', "resi "+sel) pc.show('cartoon', 'cluster1') pc.color('red', 'cluster1')
pc.save(os.path.join(save_dir, exm_p+'.pse')) print(f"file saves in {os.path.join(save_dir, exm_p+'.pse')}")
|