bed_utils.py 文件源码

def intersect(bed1, bed2, bedOut):
    if not bed2:
        shutil.copyfile(bed1,bedOut)
        return

    with open(bed1) as f:
        bed_dict1 = tk_io.get_target_regions(f)

    with open(bed2) as f:
        bed_dict2 = tk_io.get_target_regions(f)

    all_common_chroms = [chrom for chrom in bed_dict1.keys() if chrom in bed_dict2]
    bed_dict_intersect ={}

    for chrom in all_common_chroms:
        bed_dict_intersect[chrom] = bed_dict1[chrom].intersect(bed_dict2[chrom])

    writeOut(bed_dict_intersect, bedOut)