| 129 | |
| 130 | |
| 131 | def plot_sh(df, ydim="test_duration"): |
| 132 | import matplotlib.pyplot as plt |
| 133 | fig, ax = plt.subplots() |
| 134 | |
| 135 | calcs = df.harmonics_calculation.unique() |
| 136 | for calc in calcs: |
| 137 | df_calc = df.loc[df.harmonics_calculation == calc] |
| 138 | |
| 139 | df_ = df_calc.set_index("poly")[ydim].sort_index() |
| 140 | ax.plot(df_.index, df_.values, "o-") |
| 141 | |
| 142 | ax.legend(calcs) |
| 143 | ax.set_xlabel("legendre polynomials") |
| 144 | ax.set_ylabel(ydim) |
| 145 | plt.show() |
| 146 | |
| 147 | def plot_comp(df, ydim="test_duration"): |
| 148 | import matplotlib.pyplot as plt |