How do I pull a subset of an ARVIZ array within the inference data?

Thank you.

I’m following this post but with my own data: A Primer on Bayesian Methods for Multilevel Modeling — PyMC documentation

The post makes sense in walking my audience through the process. However, I would like to do the same thing, only using ArviZ the correct way. I don’t see how to exactly duplicate these plots in the documentation but will continue to try and and improve the ArviZ way.

Thanks for the help.