parse script

parse.py

+import numpy as np
+import pandas as pd
+import matplotlib.pyplot as plt
+import seaborn as sns
+
+from RNAstats import loop_counter
+
+
+def structure_info(df, sse='multi'):
+    """
+    Compute counts for a given SSE feature.
+    """
+    df[sse] = df['structure'].apply(lambda x: loop_counter(x)[sse])
+    return df
+
+def struc_plot(df, sse='multi'):
+    df = structure_info(df, sse=sse)
+    print(df)
+    strucs = df.groupby('mutations').mean()
+    sse_mean = strucs[sse]
+    sns.tsplot(sse_mean)
+    plt.show()
+
+    # print(sse_mean)
+
+def mut_plot():
+    sig = lambda e: 1 / (1 + np.exp(-e/10))
+    plt.plot([sig(x) for x in np.arange(-30, 0, 1)])
+    plt.show()
+if __name__ == "__main__":
+    path = "maternal_50_adaptive_0.csv"
+    df = pd.read_csv(path)
+    struc_plot(df, sse='multi')
+    # mut_plot()
+    pass