stanio_quickstart_json_dump_and_csv_read_numpy.py

1import numpy as np
2import stanio
3
4# 1. Prepare data to be used in a Stan model
5data = {
6    "N": 3,
7    "x": [1, 2, 3],
8    "y": np.array([4.5, 5.5, 6.5])
9}
10
11# 2. Write the data to a Stan-compatible JSON file
12stanio.dump_stan_json("data.json", data)
13
14# 3. (Optional) Read data back from a Stan JSON file
15read_data = stanio.read_stan_json("data.json")
16
17# 4. Read results from Stan CSV output files (e.g., from CmdStan)
18# This returns a dictionary of NumPy arrays
19# Note: 'output.csv' should be a path to a valid Stan CSV file
20# fit = stanio.read_stan_csv("output.csv")