Mercurial > repos > bimib > cobraxy
diff COBRAxy/ras_to_bounds.py @ 94:e844f7dab6fe draft
Uploaded
author | luca_milaz |
---|---|
date | Sun, 13 Oct 2024 11:43:08 +0000 |
parents | 7e703e546998 |
children | 54ded7f28a60 |
line wrap: on
line diff
--- a/COBRAxy/ras_to_bounds.py Sun Oct 13 11:41:34 2024 +0000 +++ b/COBRAxy/ras_to_bounds.py Sun Oct 13 11:43:08 2024 +0000 @@ -54,20 +54,20 @@ required = False, help = 'input ras') + parser.add_argument('-rn', '--names', + type=str, + help = 'ras class names') + parser.add_argument('-rs', '--ras_selector', required = True, type=utils.Bool("using_RAS"), help = 'ras selector') - - parser.add_argument('-c', '--classes', - type = str, - required = False, - help = 'input classes') parser.add_argument('-cc', '--cell_class', type = str, help = 'output of cell class') + ARGS = parser.parse_args() return ARGS @@ -128,6 +128,7 @@ scaling_factor = ras_row[reaction] lower_bound=model.reactions.get_by_id(reaction).lower_bound upper_bound=model.reactions.get_by_id(reaction).upper_bound + #warning("Reaction: "+reaction+" Lower Bound: "+str(lower_bound)+" Upper Bound: "+str(upper_bound)+" Scaling Factor: "+str(scaling_factor)) valMax=float((upper_bound)*scaling_factor) valMin=float((lower_bound)*scaling_factor) if upper_bound!=0 and lower_bound==0: @@ -190,6 +191,8 @@ if ras is not None: Parallel(n_jobs=cpu_count())(delayed(process_ras_cell)(cellName, ras_row, model, rxns_ids, output_folder) for cellName, ras_row in ras.iterrows()) + #for cellName, ras_row in ras.iterrows(): + #process_ras_cell(cellName, ras_row, model, rxns_ids, output_folder) else: model_new = model.copy() apply_ras_bounds(model_new, pd.Series([1]*len(rxns_ids), index=rxns_ids), rxns_ids) @@ -218,10 +221,10 @@ if(ARGS.ras_selector == True): ras_file_list = ARGS.input_ras.split(",") - if(len(ras_file_list)>1): - ras_class_names = [cls.strip() for cls in ARGS.classes.split(',')] - else: - ras_class_names = ["placeHolder"] + ras_file_names = ARGS.names.split(",") + ras_class_names = [] + for file in ras_file_names: + ras_class_names.append(file.split(".")[0]) ras_list = [] class_assignments = pd.DataFrame(columns=["Patient_ID", "Class"]) for ras_matrix, ras_class_name in zip(ras_file_list, ras_class_names): @@ -232,10 +235,11 @@ ras = ras.astype(float) ras_list.append(ras) for patient_id in ras.index: - class_assignments = class_assignments.append({"Patient_ID": patient_id, "Class": ras_class_name}, ignore_index=True) + class_assignments.loc[class_assignments.shape[0]] = [patient_id, ras_class_name] + # Concatenate all ras DataFrames into a single DataFrame - ras_combined = pd.concat(ras_list, axis=1) + ras_combined = pd.concat(ras_list, axis=0) # Normalize the RAS values by max RAS ras_combined = ras_combined.div(ras_combined.max(axis=0)) ras_combined = ras_combined.fillna(0) @@ -261,8 +265,7 @@ if(ARGS.ras_selector == True): generate_bounds(model, medium, ras = ras_combined, output_folder=ARGS.output_folder) - if(len(ras_list)>1): - class_assignments.to_csv(ARGS.cell_class, sep = '\t', index = False) + class_assignments.to_csv(ARGS.cell_class, sep = '\t', index = False) else: generate_bounds(model, medium, output_folder=ARGS.output_folder)