view PDAUG_AddClassLabel/PDAUG_AddClassLabel.py @ 2:bc6aef6ec2d4 draft

"planemo upload for repository https://github.com/jaidevjoshi83/pdaug commit 8b18552f6d2b2261efebe1075ff4c18a295b94dd"
author jay
date Tue, 29 Dec 2020 18:38:48 +0000
parents 7557b48b2872
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)