From b4c9ee1186d52a304e46c525f88f21c5d4e0c8c8 Mon Sep 17 00:00:00 2001 From: chaopower Date: Thu, 11 Jan 2024 16:50:33 +0800 Subject: [PATCH] fusion bug --- codes/postprocess.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/codes/postprocess.py b/codes/postprocess.py index 909ac9a..cee66c1 100755 --- a/codes/postprocess.py +++ b/codes/postprocess.py @@ -286,17 +286,17 @@ class PostProcess: filter_sum_pos_df['Validated'] = 1 filter_sum_pos_df = filter_sum_pos_df.fillna('.') - grouped_df = filter_sum_pos_df.groupby(['POS', 'REF', 'ALT']) + grouped_df = filter_sum_pos_df.groupby(['POS', 'REF', 'FUSION']) # 对每个分组进行操作 for group_name, group_data in grouped_df: pos, ref, alt = group_name if any(group_data['AMP_mut_level'] == 'I'): filter_condition = (filter_sum_pos_df['POS'] == pos) & \ (filter_sum_pos_df['REF'] == ref) & \ - (filter_sum_pos_df['ALT'] == alt) + (filter_sum_pos_df['FUSION'] == alt) filter_sum_pos_df.loc[filter_condition, 'AMP_mut_level'] = 'I' - pos_dict = filter_sum_pos_df.set_index(['POS', 'REF', 'ALT'])['AMP_mut_level'].to_dict() + pos_dict = filter_sum_pos_df.set_index(['POS', 'REF', 'FUSION'])['AMP_mut_level'].to_dict() filter_sum_pos_res = filter_sum_pos_df.to_dict('records') filter_sum = os.path.join(self.path, 'fusion', @@ -309,7 +309,7 @@ class PostProcess: filter_sum_df['Validated'] = 1 level_dict = dict() level_dict.update(pos_dict) - filter_sum_df['AMP_mut_level'] = filter_sum_df.set_index(['POS', 'REF', 'ALT']).index.map(level_dict) + filter_sum_df['AMP_mut_level'] = filter_sum_df.set_index(['POS', 'REF', 'FUSION']).index.map(level_dict) filter_sum_df = filter_sum_df.fillna('.') filter_sum_res = filter_sum_df.to_dict('records')