I use my own function
def get_stats(cmdstan_data):
# include mean and hpd
stats = az.summary(cmdstan_data,credible_interval=0.95).reset_index().rename(columns={'index':'var'})
stats['time'] = stats['var'].apply(lambda st: st[st.find("[")+1:st.find("]")])
stats['time'] = ['NA' if "[" not in y else int(x)+1 for x,y in zip(stats['time'],stats['var'])]
stats['var'] = stats['var'].apply(lambda st: st[:st.find("[")] if "[" in st else st)
return stats
it can generalized to 2d if necessary. Maybe it would be helpful for you too