import pandas as pd import sys, argparse, os men = pd.read_table(sys.argv[1],header=None) vote = pd.read_table(sys.argv[2],low_memory=False) mut = pd.read_table(sys.argv[3],header=None) merged_df = pd.merge(vote, men, how='inner', left_on=['CHROM','POS'], right_on = [0,1]) 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)