def plot_p123_gc(data, graph_name='p123_gc.png', use_p3_as_x=False, **kwargs): """Output a scatter plot of p1,p2,p3 vs gc content Expects data as array with rows as GC, P1, P2, P3 p1=blue, p2=green, p3=red """ #unpack common x axis, and decide on series names if use_p3_as_x: series_names = ['$P_1$', '$P_2$'] colors=['b','g'] x_label='$P_3$' y_label='$P_{12}$' xy_pairs = [data[3], data[1], data[3], data[2]] else: series_names = ['$P_1$', '$P_2$', '$P_3$'] colors=['b','g','r'] x_label='GC' y_label='$P_{123}$' xy_pairs = [data[0], data[1], data[0], data[2], data[0], data[3]] #plot points and write graph plot_scatter(xy_pairs, graph_grid='/',x_label=x_label,y_label=y_label, series_names=series_names, prob_axes=True, **kwargs) savefig(graph_name)
def plot_p12_p3(data, graph_name='p12_p3.png', **kwargs): """Outputs a P12 versus P3 scatter graph, optionally including regression. expects data as [P3_1, P12_1, P3_2, P12_2, ...n ]. """ plot_scatter(data, graph_shape='sqr', graph_grid='/',\ x_label="$P_3$",y_label="$P_{12}$", prob_axes=True, **kwargs) savefig(graph_name)
def plot_cai_p3_scatter(data, graph_name='cai_p3_scat.png', **kwargs): """Outputs a CAI vs P3 scatter plot. expects data as ([P3s_1, CAIs_1, P3s_2, CAIs_2, ...]) """ plot_scatter(data, graph_shape='sqr', graph_grid=None,\ x_label="$P_3$",y_label="CAI", prob_axes=True,**kwargs) savefig(graph_name)