Added PROTEINS column for maxquant

wilhelm-lab · Apr 29, 2024 · c7e3cd6 · c7e3cd6
1 parent 5eeec06
commit c7e3cd6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/spectrum_io/search_result/maxquant.py b/spectrum_io/search_result/maxquant.py
@@ -62,6 +62,7 @@ def read_result(self, tmt_labeled: str) -> pd.DataFrame:
                 "MASS",  # = Calculated Precursor mass; TODO get column with experimental Precursor mass instead
                 "SCORE",
                 "REVERSE",
+                "PROTEINS",
             ],
             sep="\t",
         )
@@ -117,6 +118,7 @@ def update_columns_for_prosit(df: pd.DataFrame, tmt_labeled: str) -> pd.DataFram
             df["MODIFIED_SEQUENCE"] = maxquant_to_internal(df["MODIFIED_SEQUENCE"].to_numpy())
         df["SEQUENCE"] = internal_without_mods(df["MODIFIED_SEQUENCE"])
         df["PEPTIDE_LENGTH"] = df["SEQUENCE"].apply(lambda x: len(x))
+        df["PROTEINS"].fillna("UNKNOWN", inplace=True)
 
         return df