def compute_dists(index):
ntcr = sample_tcrs.loc[index, ] #tcrr
edists = []
for tindex, tcr in epigroup.iterrows():
if tcr['tcr_info'] == ntcr['tcr_info']:
continue
edists.append(
ch_tcr_distances.compute_distance(tcr['tcr_info'],
ntcr['tcr_info'],
chains,
distance_params,
rep_dists=rep_dists))
edists.sort()
for nbrdist_percentile in nbrdist_percentiles:
nbrdist = ch_tcr_distances.sort_and_compute_nbrdist_from_distances(
edists, nbrdist_percentile, dont_sort=True)
wtd_nbrdist = ch_tcr_distances.sort_and_compute_weighted_nbrdist_from_distances(
edists, nbrdist_percentile, dont_sort=True)
ntcr['{}_{}_nbrdist{}'.format(
epitope, chains,
nbrdist_percentile)] = '{:.3f}'.format(nbrdist)
ntcr['{}_{}_wtd_nbrdist{}'.format(
epitope, chains,
nbrdist_percentile)] = '{:.3f}'.format(wtd_nbrdist)
return ntcr
def compute_dists(index):
ntcr = all_tcrs.loc[index, ]
edists = []
for tindex, tcr in epigroup.iterrows():
if tcr['tcr_info'] == ntcr['tcr_info'] or tindex == index:
#if epitope=='LLWNGPMAV':
# print(tcr['tcr_info'], ntcr['tcr_info'], 'continued')
continue
edists.append(
ch_tcr_distances.compute_distance(tcr['tcr_info'],
ntcr['tcr_info'],
chains,
distance_params,
rep_dists=rep_dists))
if edists[-1] == 0:
print(tcr['tcr_info'], ntcr['tcr_info'], 'added to dists')
edists.sort()
for nbrdist_percentile in nbrdist_percentiles:
nbrdist = ch_tcr_distances.sort_and_compute_nbrdist_from_distances(
edists, nbrdist_percentile, dont_sort=True)
wtd_nbrdist = ch_tcr_distances.sort_and_compute_weighted_nbrdist_from_distances(
edists, nbrdist_percentile, dont_sort=True)
ntcr['{}_{}_nbrdist{}'.format(
epitope, chains, nbrdist_percentile)] = '{:.3f}'.format(nbrdist)
ntcr['{}_{}_wtd_nbrdist{}'.format(
epitope, chains,
nbrdist_percentile)] = '{:.3f}'.format(wtd_nbrdist)
return ntcr
def compute_dists(rindex):
random_tcr = random_tcrs.loc[rindex, ]
edists = []
for tindex, tcr in epigroup.iterrows():
edists.append(ch_tcr_distances.compute_distance(tcr['tcr_info'], random_tcr['tcr_info'],
chains, distance_params, rep_dists=rep_dists))
edists.sort()
for nbrdist_percentile in nbrdist_percentiles:
nbrdist = ch_tcr_distances.sort_and_compute_nbrdist_from_distances(edists, nbrdist_percentile,
dont_sort=True)
wtd_nbrdist = ch_tcr_distances.sort_and_compute_weighted_nbrdist_from_distances(edists,
nbrdist_percentile,
dont_sort=True)
random_tcr['{}_{}_nbrdist{}'.format(epitope, chains, nbrdist_percentile)] = '{:.3f}'.format(
nbrdist)
random_tcr['{}_{}_wtd_nbrdist{}'.format(epitope, chains, nbrdist_percentile)] = '{:.3f}'.format(
wtd_nbrdist)
return random_tcr
def compute_dists(rindex):
random_tcr = random_tcrs.loc[rindex, ]
edists = []
for tindex, tcr in epigroup.iterrows():
edists.append(ch_tcr_distances.compute_distance(tcr['tcr_info'], random_tcr['tcr_info'],
chains, distance_params, rep_dists=rep_dists))
edists.sort()
for nbrdist_percentile in nbrdist_percentiles:
nbrdist = ch_tcr_distances.sort_and_compute_nbrdist_from_distances(edists, nbrdist_percentile,
dont_sort=True)
wtd_nbrdist = ch_tcr_distances.sort_and_compute_weighted_nbrdist_from_distances(edists,
nbrdist_percentile,
dont_sort=True)
random_tcr['{}_{}_nbrdist{}'.format(epitope, chains, nbrdist_percentile)] = '{:.3f}'.format(
nbrdist)
random_tcr['{}_{}_wtd_nbrdist{}'.format(epitope, chains, nbrdist_percentile)] = '{:.3f}'.format(
wtd_nbrdist)
return random_tcr