How to plot beta[363] with `az.plot_trace`

That is right. Moreover, if you define dimension or coordinate names you can use them as coords like in this example.

This blog post: https://oriolabrilpla.cat/en/blog/posts/2022/pymc-arviz.html might be helpful in navigating coords, dims and arviz-pymc integration

2 Likes