chaopower
/
pipeline
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fusion bug

master
chaopower 2024-01-11 16:50:33 +08:00
parent befe5b7693
commit b4c9ee1186
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
codes/postprocess.py
View File

@ -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')