# Constructed data: 10 wild-type and 10 knockout mice; the knockout does not change freezing or extinction.
# Python 3.13, NumPy, pandas. Seed 1, the first one tried.
import numpy as np, pandas as pd
from model import study, SECONDS
rng = np.random.default_rng(1)
rows = [(g, f"{g[0].upper()}{m}", t + 1, s, SECONDS) for g, m, x in study(rng) for t, s in enumerate(x)]
pd.DataFrame(rows, columns=["genotype", "mouse", "tone", "seconds_freezing", "seconds_scored"]).to_csv("freezing.csv", index=False)