From 35f1f86844f46d2633d899c4e9f6a56f3ddf0706 Mon Sep 17 00:00:00 2001 From: zsloan Date: Tue, 16 Feb 2021 22:07:14 +0000 Subject: Changed a couple loops to checking if value is in a list comprehension --- wqflask/wqflask/marker_regression/display_mapping_results.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/wqflask/wqflask/marker_regression/display_mapping_results.py b/wqflask/wqflask/marker_regression/display_mapping_results.py index 4df56190..47c64a69 100644 --- a/wqflask/wqflask/marker_regression/display_mapping_results.py +++ b/wqflask/wqflask/marker_regression/display_mapping_results.py @@ -1603,9 +1603,8 @@ class DisplayMappingResults(object): for k, _geno in enumerate (_chr[j].genotype): plotbxd=0 - for item in smd: - if item.name == samplelist[k]: - plotbxd=1 + if samplelist[k] in [item.name for item in smd]: + plotbxd=1 if (plotbxd == 1): ind = 0 @@ -1614,6 +1613,7 @@ class DisplayMappingResults(object): counter = counter + 1 if item.name == samplelist[k]: ind = counter + maxind=max(ind, maxind) # lines @@ -1687,9 +1687,8 @@ class DisplayMappingResults(object): for j, _geno in enumerate(_chr[1].genotype): plotbxd=0 - for item in smd: - if item.name == samplelist[j]: - plotbxd=1 + if samplelist[j] in [item.name for item in smd]: + plotbxd=1 if (plotbxd == 1): -- cgit v1.2.3