Add cyto and nuc RNA columns for total RNA in data_script.py

MunskyGroup · Oct 7, 2024 · 83952fd · 83952fd
1 parent f8e8027
commit 83952fd
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/WorkSpace/GregorNeuert/data_script.py b/WorkSpace/GregorNeuert/data_script.py
@@ -6,15 +6,18 @@
 
 # Filter the data based on Condition = '2M_NaCl_Step'
 filtered_data = data.loc[data['Condition'].str.contains('2M_NaCl_Step', case=False, na=False), 
-                         ['Cell_id', 'Condition', 'Replica', 'Conc_mM', 'Time_index_min', 'RNA_STL1_cyto_TS3Full']]
+                         ['Cell_id', 'Condition', 'Replica', 'Conc_mM', 'Time_index_min', 
+                          'RNA_STL1_cyto_TS3Full', 'RNA_STL1_nuc_TS3Full']]
 
-filtered_data.rename(columns={'Time_index_min': 'time'}, inplace=True)
+# Sum 'RNA_STL1_cyto_TS3Full' and 'RNA_STL1_nuc_TS3Full' into new column 'RNA_STL1_total_TS3Full'
+filtered_data['RNA_STL1_total_TS3Full'] = filtered_data['RNA_STL1_cyto_TS3Full'] + filtered_data['RNA_STL1_nuc_TS3Full']
 
+# Replace 'Time_index_min' with 'time' to be read by SSIT.m
+filtered_data.rename(columns={'Time_index_min': 'time'}, inplace=True)
 
 # Define the output file path
 output_file = 'filtered_data_2M_NaCl_Step.csv'
 
-
 # Save the filtered data to a new CSV file
 filtered_data.to_csv(output_file, index=False)