Mercurial > repos > jay > pdaug_peptide_structure_builder

comparison PDAUG_Merge_Dataframes/PDAUG_Merge_Dataframes.py @ 4:a6322865e23a draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
"planemo upload for repository https://github.com/jaidevjoshi83/pdaug commit 45ebf32dcaa1eed91670d3a2491f9cf3dfb535ef"
author jay
date Tue, 12 Jan 2021 19:07:03 +0000
parents 0118cc553f3b
children
comparison
equal deleted inserted replaced
3:2b05b50ca410 4:a6322865e23a
1 import glob 1 import glob
2 import pandas as pd 2 import pandas as pd
3 import sys 3 import sys
4 4
5 files = sys.argv[1] 5
6 out_file = sys.argv[2] 6 def MergeData(infiles, add_class_label, class_label, OutPut):
7
8 data_frame = pd.DataFrame()
9
10 if add_class_label == 'True' or add_class_label == 'true':
11 for i, file in enumerate(infiles.split(',')):
12 df1 = pd.read_csv(file,sep='\t')
13 df2 = pd.DataFrame(df1.shape[0]*[i], columns=[class_label])
14 df3 = pd.concat([df1,df2], axis=1)
15 data_frame = pd.concat([data_frame,df3])
16 final_DF = data_frame.fillna(0)
17
18 else:
19
20 for file in infiles.split(','):
21 df1 = pd.read_csv(file,sep='\t')
22 data_frame = pd.concat([data_frame,df1])
23 final_DF = data_frame.fillna(0)
24
25 final_DF.to_csv(OutPut, sep="\t", index=False)
7 26
8 27
9 data_frame = pd.read_csv(files.split(',')[0],sep='\t') 28 if __name__=="__main__":
10 29
30 import argparse
31 parser = argparse.ArgumentParser()
32 parser.add_argument("-I", "--infiles", required=True, default=None, help=".tsv")
33 parser.add_argument("-L", "--add_class_label", required=False, default=False, help="Path to target tsv file")
34 parser.add_argument("-C", "--class_label", required=False, default='class_label', help="Path to target tsv file")
35 parser.add_argument("-O", "--OutPut", required=False, default='Out.tsv', help="Path to target tsv file")
11 36
12 for file in files.split(',')[1:]: 37 args = parser.parse_args()
13 38
14 df1 = pd.read_csv(file,sep='\t') 39 MergeData(args.infiles, args.add_class_label, args.class_label, args.OutPut)
15 data_frame = pd.concat([data_frame,df1])
16
17 final_DF = data_frame.fillna(0)
18
19 final_DF.to_csv(out_file,sep="\t", index=False)
20
21
22
23
24
25
26
27
28