1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
"""Parsers for generating some files in genenetwork"""
import os
from typing import Any, Dict, List, Tuple
def parse_genofile(file_path: str) -> Tuple[List[str],
List[Dict[str, Any]]]:
"""Parse a genotype file with a given format"""
if not os.path.exists(file_path):
raise FileNotFoundError
__map = {
'b': -1,
'd': 1,
'h': 0,
'u': None,
}
genotypes, strains = [], []
with open(file_path, "r") as _genofile:
for line in _genofile:
line = line.strip()
if line.startswith(("#", "@")):
continue
cells = line.split()
if line.startswith("Chr"):
strains = cells[4:]
strains = [strain.lower() for strain in strains]
continue
values = [__map.get(value.lower(), None) for value in cells[4:]]
genotype = {
"chr": cells[0],
"locus": cells[1],
"cm": cells[2],
"mb": cells[3],
"values": values,
"dicvalues": dict(zip(strains, values)),
}
genotypes.append(genotype)
return strains, genotypes
|
