Mercurial > repos > jay > pdaug_peptide_structure_builder
view PDAUG_AddClassLabel/PDAUG_AddClassLabel.py @ 8:955d6f80fa61 draft
"planemo upload for repository https://github.com/jaidevjoshi83/pdaug commit d396d7ff89705cc0dd626ed32c45a9f4029b1b05"
author | jay |
---|---|
date | Mon, 10 Jan 2022 04:33:39 +0000 |
parents | 0118cc553f3b |
children |
line wrap: on
line source
import pandas as pd import argparse parser = argparse.ArgumentParser() parser.add_argument("-I", "--InFile", required=True, default=None, help="Input data file") parser.add_argument("-C", "--ClassLabel", required=False, default=0, help="Class Label 0 or 1") parser.add_argument("-t", "--ClassLabelTitle", required=False, default='Class_label', help="Title to use for class label column (Class_label)") parser.add_argument("-O", "--OutFile", required=False, default='OutFile.tsv', help="Output file name") args = parser.parse_args() df1 = pd.read_csv(args.InFile, sep="\t") df2 = pd.DataFrame([args.ClassLabel]*df1.shape[0], columns=[args.ClassLabelTitle]) df = pd.concat([df1, df2], axis=1) df.to_csv(args.OutFile, sep="\t", index=False)