view median_size.py @ 6:7585f18653bc draft

Uploaded
author estrain
date Fri, 10 Mar 2023 15:16:21 +0000
parents a53acd38d77e
children
line wrap: on
line source

#!/usr/bin/env

## Errol Strain (estrain@gmail.com)
## calculate median insert size from sam file

import numpy as np

def get_data(infile):
    lengths = []
    for line in infile:
        if line.startswith('@'):
            pass
        else:
            line = line.rsplit()
            length = int(line[8])
            if length > 0:
                lengths.append(length)
            else:
                pass
    return lengths

if __name__ == "__main__":
    import sys
    lengths = get_data(sys.stdin)
    md = int(np.median(lengths))
print(md)