tpm_normalize_col.py

def tpm_normalize_col(col, data):
    "TPM normalize a column of our counts data"
    rpk = np.divide(col, data["Length"])
    scale_factor = np.divide(np.sum(rpk), 1e6)
    tpm = np.divide(rpk, scale_factor)
    return tpm


tpm_seqdata = seqdata.apply(
    lambda x: tpm_normalize_col(x, seqdata)
    if x.name in sample_cols
    else x
)