aboutsummaryrefslogtreecommitdiff
path: root/gn3/utility/get_group_samplelists.py
diff options
context:
space:
mode:
Diffstat (limited to 'gn3/utility/get_group_samplelists.py')
-rw-r--r--gn3/utility/get_group_samplelists.py47
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