python - emcee.EnsembleSampler - 循环的构造

Question

我的 MCMC 脚本中有这个命令，它给出了位置、概率和状态。如何重写此命令以循环以连续获取值并将它们保存到文件（而不是我可以看到当前结果）而不是整个列表，好吗？

sampler = emcee.EnsembleSampler(nwalkers, ndim, lnprob, backend=backend)

pos, prob, state = sampler.run_mcmc(p0, niter, progress=True)


#  for i in range(1,3):
#    pos = i  
#    file_object = open('pos.txt', 'a')
#    file_object.write(str(pos) + '\n')
#   file_object.close()

#   file_object = open('prob.txt', 'a')
#   file_object.write(str(prob) + '\n')
#   file_object.close()