diff options
Diffstat (limited to 'gn3/utility/get_group_samplelists.py')
-rw-r--r-- | gn3/utility/get_group_samplelists.py | 47 |
1 files changed, 0 insertions, 47 deletions
diff --git a/gn3/utility/get_group_samplelists.py b/gn3/utility/get_group_samplelists.py deleted file mode 100644 index 8fb322a..0000000 --- a/gn3/utility/get_group_samplelists.py +++ /dev/null @@ -1,47 +0,0 @@ - -"""module for group samplelist""" -import os - -#todo close the files after opening -def get_samplelist(file_type, geno_file): - """get samplelist function""" - if file_type == "geno": - return get_samplelist_from_geno(geno_file) - elif file_type == "plink": - return get_samplelist_from_plink(geno_file) - -def get_samplelist_from_geno(genofilename): - if os.path.isfile(genofilename + '.gz'): - genofilename += '.gz' - genofile = gzip.open(genofilename) - else: - genofile = open(genofilename) - - for line in genofile: - line = line.strip() - if not line: - continue - if line.startswith(("#", "@")): - continue - break - - headers = line.split("\t") - - if headers[3] == "Mb": - samplelist = headers[4:] - else: - samplelist = headers[3:] - return samplelist - - - -def get_samplelist_from_plink(genofilename): - """get samplelist from plink""" - genofile = open(genofilename) - - samplelist = [] - for line in genofile: - line = line.split(" ") - samplelist.append(line[1]) - - return samplelist |