Name the split models with 3 digits instead of 1

pkoukos · Mar 16, 2017 · f129382 · f129382
1 parent 00f2bbc
commit f129382
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pdb_splitmodel.py b/pdb_splitmodel.py
@@ -59,6 +59,7 @@ def _extract_models(fhandle):
             output_handle.write(line)
         elif line.startswith('MODEL'):
             model_no = line.split()[-1]
+            model_no = '{0:03d}'.format(int(model_no))
             output_handle = open(fname_root + '_' + model_no + '.pdb', 'w')
         elif line.startswith('ENDMDL'):
             output_handle.write('END\n')