21 lines
626 B
Python
21 lines
626 B
Python
from matplotlib import pyplot as plt
|
|
|
|
from lib.lib import read_sp, aggregate_sp_fd
|
|
|
|
data_2d_sp = read_sp("../data/rust-stick-probability")
|
|
data_2d_sp_low_p = read_sp("../data/rust-sp-low-p")
|
|
sp_fd_data = aggregate_sp_fd(data_2d_sp)
|
|
|
|
# %%
|
|
plt.fill_between(sp_fd_data.index, sp_fd_data.fd - sp_fd_data.fd_std, sp_fd_data.fd + sp_fd_data.fd_std, alpha=0.2, label=f"Standard error band")
|
|
plt.plot(sp_fd_data.index, sp_fd_data.fd, color='tab:blue', label='NF, fd mean, seeds = 100')
|
|
|
|
plt.xlabel("$p_{stick}$")
|
|
plt.ylabel("$fd$")
|
|
plt.legend()
|
|
|
|
# %%
|
|
plt.savefig('../figures/sp-fd.svg')
|
|
plt.savefig('../figures/sp-fd.png')
|
|
plt.show()
|