如何为GetDist情节保存主持人chian

2024-09-30 08:26:54 发布

您现在位置:Python中文网/ 问答频道 /正文

我想和主持人做一个简单的练习。代码运行后,我可以得到一个绘图,但现在我想保存该链,以便使用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服务或绘制链吗? 先谢谢你


Tags: log绘图returndefnpdifftemplike

热门问题