import pandas as pd import sys, argparse, os men = pd.read_table(sys.argv[1],header=None,low_memory=False) vote = pd.read_table(sys.argv[2],low_memory=False) men[1]=men[1].astype(str) merged_df = pd.merge(vote, men, how='inner', left_on=['CHROM','POS'], right_on = [0,1]) lcl5_dat = merged_df[(merged_df[5]!='./.') & (merged_df[5]!='0/0')] merged_df['mendelian_check'] = 'MIE' merged_df.loc[merged_df[2]=='1:1:1','mendelian_check'] = 'MP' sub = merged_df[['CHROM','POS','LCL5_detected_num','mendelian_check',2]] sub.columns=['CHROM','POS','detected_num','mendelian','detail'] genotype_sub = pd.merge(sub, mut, how='inner', left_on=["CHROM","POS"], right_on = [0,1]) genotype_sub = genotype_sub[['CHROM','POS','detected_num','mendelian','detail',2,3]] genotype_sub.to_csv(sys.argv[4],header=0,sep="\t",index=0)