Угловой сюжет добавить еще одну точку - PullRequest
Новая
       51

Угловой сюжет добавить еще одну точку

0 голосов
/ 16 апреля 2020

Я использую следующее MWE: 

def contour(fit, mu_des,dt_des):
    la = fit.extract(permuted=True)
    mu_pos = np.vstack((la['mu'])).T
    dt_pos = np.vstack((la['dt'])).T
    postsamples = np.vstack((mu_pos, dt_pos)).T
    fig = corner.corner(postsamples, labels=[r"$\mu$", r"$dt$"],color='g', smooth=1,
                        levels=(0.68,0.95),plot_density=0, fill_contours=1, plot_datapoints=1,
                           truths=list([mu_des,dt_des]))
    return fig

Теперь я также хочу показать пользовательскую точку на графике. Вопрос чем-то похож на этот вопрос , но я не мог последовать за ответом. Пожалуйста, помогите.

1 Ответ

0 голосов
/ 17 апреля 2020

Хорошо, я получил ответ по ссылке , приведенной ниже. Можно изменить функцию как 

def contour(fit):
    la = fit.extract(permuted=True)
    mu_pos = np.vstack((la['mu'])).T
    dt_pos = np.vstack((la['dt'])).T
    #print(mu_pos.shape)
    #P_pos=np.vstack((la['P'])).T
    postsamples = np.vstack((mu_pos, dt_pos)).T
    #print(postsamples.shape)
    fig = corner.corner(postsamples, labels=[r"$\mu$", r"$dt$"],color='g', smooth=1,
                        levels=(0.68,0.95),plot_density=0, fill_contours=1, plot_datapoints=1,
                           truths=list([mu_des,dt_des]))
    axes = np.array(fig.axes).reshape((2, 2)); print(axes)
    ax = axes[1, 0]; 
    ax.plot(value1, value2, "sg")
    return fig
...