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)
