def plot_seqlogo(ax, seq_1hot, sat_score_ti, pseudo_pct=0.05): """ Plot a sequence logo for the loss/gain scores. Args: ax (Axis): matplotlib axis to plot to. seq_1hot (Lx4 array): One-hot coding of a sequence. sat_score_ti (L_sm array): Minimum mutation delta across satmut length. pseudo_pct (float): % of the max to add as a pseudocount. """ satmut_len = len(sat_score_ti) # add pseudocounts sat_score_ti += pseudo_pct * sat_score_ti.max() # expand sat_score_4l = expand_4l(sat_score_ti, seq_1hot) plots.seqlogo(sat_score_4l, ax)
def plot_seqlogo(ax, seq_align, sat_score_ti, pseudo_pct=0.05): """ Plot a sequence logo for the loss/gain scores. Args: ax (Axis): matplotlib axis to plot to. seq_align (L array): Sequence nucleotides, with gaps. sat_score_ti (L_sm array): Minimum mutation delta across satmut length. pseudo_pct (float): % of the max to add as a pseudocount. """ satmut_len = len(sat_score_ti) # add pseudocounts sat_score_ti += pseudo_pct * np.nanmax(sat_score_ti) # expand sat_score_4l = expand_4l(sat_score_ti, seq_align) plots.seqlogo(sat_score_4l, ax)