python_snippets

def remove_description_from_fasta(input, output, regex):
    with open(input,) as input_stream, open(output , 'w') as output_stream:
        for line in input_stream:                                                         
    if line.startswith('>'):
        if regex in line:
            output_stream.write(line)
            write = True 
        else:
            write = False
    elif write:
        output_stream.write(line)