Sampling a specific country from the hierarchical model example

Getting log radon from a and b is just addition:

a_samples = hierarchical_trace.get_values('a', thin=5)
b_samples = hierarchical_trace.get_values('b', thin=5)
aitkin_idx = 0  # this is a guess; the point is, each county has a unique integer index
aitkin_a_samples = a_samples[:, aitkin_idx]
aitkin_b_samples = b_samples[:, aitkin_idx]

Posterior distribution of mean log radon in Aitkin, for houses w/ and w/o basements:

aitkin_log_radon_w_basement = aitkin_a_samples 
aitkin_log_radon_wo_basement = aitkin_a_samples + aitkin_b_samples
1 Like