我想和主持人做一个简单的练习。代码运行后,我可以得到一个绘图,但现在我想保存该链,以便使用Getdist进行绘图。我附加了一个最小的代码
def log_like(theta,z,mu):
omegaL,omegaM=theta
for i in range(0,len(z)):
arrayofz[i]=dl(omegaL,omegaM,z[i])
arrayofz[i]=eta+(5*np.log10(arrayofz[i]))
diff=mu - arrayofz
temp=-0.5*(np.sum((diff**2)/sigma**2))
return temp
def log_prior(theta):
omegaL,omegaM=theta
if 0<=omegaL <= 1 and 0<=omegaM <= 1:
return 0.0
return -np.inf
def log_probability(theta,z,mu):
return log_prior(theta)+log_like(theta,z,mu)
N=50
goodguess=np.array([0.7,0.3])
pos = goodguess + (np.random.randn(N, 2)/10)
nwalkers=N
ndim=2
#---------------------------------
sampler = emcee.EnsembleSampler(nwalkers, ndim, log_probability, args=(z,mu))
sampler.run_mcmc(pos, 300, progress=True);
samples = sampler.get_chain()
labels = ["omegaL", "omegaM"]
flat_samples = sampler.get_chain(discard=10, flat=True)`
如果我使用角点,我可以得到一个情节(带有切割轴,但这是另一个故事)。 我试图将示例保存为.txt,但什么也没有。 你知道如何使用GetDist服务或绘制链吗? 先谢谢你
目前没有回答
相关问题 更多 >
编程相关推荐