From 7a3e0fdac498a9d22c6ea8789b42c5c882120e00 Mon Sep 17 00:00:00 2001 From: Frederick Muriuki Muriithi Date: Fri, 16 Feb 2024 16:34:11 +0300 Subject: Convert missing value codes to None --- r_qtl/r_qtl2.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/r_qtl/r_qtl2.py b/r_qtl/r_qtl2.py index e4c1894..4f0ab2a 100644 --- a/r_qtl/r_qtl2.py +++ b/r_qtl/r_qtl2.py @@ -306,3 +306,10 @@ def raw_file_data(zipfilepath: Union[str, Path], def strip_comments(rawdata: Iterator[str], commentchar) -> Iterator[str]: """Remove comments from raw text.""" return (line for line in rawdata if not line.startswith(commentchar)) + +def missing_value_codes_to_none(value: str, + nastrings: tuple[str, ...]) -> Optional[str]: + """ + If 'value' is a missing value code, return `None`, otherwise return 'value'. + """ + return value if value not in nastrings else None -- cgit v1.2.3