import pymol from pymol import cmd cmd.fetch('1mdm') cmd.select("set1", "resn D* and name O*'") cmd.select("set2", "name OP*") cmd.select("set3", "resn D* and name N*")