import argparse parser = argparse.ArgumentParser() parser.add_argument('-i', help = "input PDB file", required = True, dest = 'pdb_file') args = parser.parse_args() pdb_file = args.pdb_file pdb_file_opened = open(pdb_file, 'r') temp_factors = list() resids_in_dif_states = dict() for line in pdb_file_opened: if line.startswith('ATOM'): temp_factor = float(line[60:66]) fill_factor = float(line[56:60]) resid_number = int(line[23:26]) temp_factors.append(temp_factor) if fill_factor < 1.00: resids_in_dif_states[resid_number] = 'value' temp_factors.sort() mediane = len(temp_factors)/2 print temp_factors[mediane] resids = resids_in_dif_states.keys() resids.sort() print resids, len(resids) pdb_file_opened.close()