|
7 | 7 | import polars as pl |
8 | 8 |
|
9 | 9 | HERE = Path(__file__).parent |
| 10 | +CSV_FILE_METHIONINE = HERE / "methionine.csv" |
10 | 11 | CSV_FILE_ROSENBROCK = HERE / "rosenbrock.csv" |
11 | | -CSV_FILE_LINEAR = HERE / "linear_pathway.csv" |
| 12 | +CSV_FILE_LINEAR = HERE / "linear.csv" |
12 | 13 | CSV_FILE_TRAJECTORY = HERE / "trajectory.csv" |
13 | 14 |
|
14 | 15 |
|
@@ -169,12 +170,15 @@ def trajectory_fig(result: pl.DataFrame): |
169 | 170 |
|
170 | 171 | def main(): |
171 | 172 | matplotlib.rcParams["savefig.dpi"] = 300 |
172 | | - df_rb = pl.read_csv(CSV_FILE_ROSENBROCK) |
173 | | - df_linear = pl.read_csv(CSV_FILE_LINEAR) |
| 173 | + df_methionine = pl.read_csv(CSV_FILE_METHIONINE).with_columns( |
| 174 | + model=pl.lit("Methionine"), dim=0 |
| 175 | + ) |
| 176 | + df_rb = pl.read_csv(CSV_FILE_ROSENBROCK).with_columns(model=pl.lit("Rosenbrock")) |
| 177 | + df_linear = pl.read_csv(CSV_FILE_LINEAR).with_columns( |
| 178 | + model=pl.lit("Small enzyme network"), dim=0 |
| 179 | + ) |
174 | 180 | df_trajectory = pl.read_csv(CSV_FILE_TRAJECTORY) |
175 | | - df_rb = df_rb.with_columns(model=pl.lit("Rosenbrock")) |
176 | | - df_linear = df_linear.with_columns(model=pl.lit("Small enzyme network"), dim=0) |
177 | | - df_performance = pl.concat([df_rb, df_linear], how="align") |
| 181 | + df_performance = pl.concat([df_methionine, df_rb, df_linear], how="align") |
178 | 182 |
|
179 | 183 | f, _ = performance_fig(df_performance) |
180 | 184 | f.savefig(HERE / "performance.png", bbox_inches="tight") |
|
0 commit comments