solution, plot_kwargs = subdata['solution'], subdata['plot_kwargs']
ax3.plot(solution.t, solution.cz, label=label, **plot_kwargs)
ax3.set_xlim(xlim)
ax3.set_ylim(ylim)
if args.extra_data:
# Add the force coefficients from Li and Dong (2016).
# Data were digitized from Figure 9 of the article.
scatter_kwargs = dict(s=10, facecolors='none', edgecolors='black')
ax1.scatter(*rodney.li_dong_2016_load_ct(),
label='Li & Dong (2016)',
**scatter_kwargs)
ax2.scatter(*rodney.li_dong_2016_load_cl(),
label='Li & Dong (2016)',
**scatter_kwargs)
ax3.scatter(*rodney.li_dong_2016_load_cz(),
label='Li & Dong (2016)',
**scatter_kwargs)
ax1.legend(frameon=False,
prop=dict(size=10),
scatterpoints=3,
labelspacing=0.25)
fig.tight_layout()
if args.save_figures:
figdir = maindir / 'figures'
figdir.mkdir(parents=True, exist_ok=True)
filepath = figdir / 'force_coefficients_compare_dx.png'
fig.savefig(filepath, dpi=300, bbox_inches='tight')
ax3.set_xlabel('$t / T$')
ax3.set_ylabel('$C_Z$')
for label, subdata in data.items():
solution, plot_kwargs = subdata['solution'], subdata['plot_kwargs']
ax3.plot(solution.t, solution.cz, label=label, **plot_kwargs)
ax3.set_xlim(xlim)
ax3.set_ylim(ylim)
if args.extra_data:
# Add the force coefficients from Li and Dong (2016).
# Data were digitized from Figure 9 of the article.
scatter_kwargs = dict(s=10, facecolors='none', edgecolors='black')
ax1.scatter(*rodney.li_dong_2016_load_ct(), label='Li & Dong (2016)',
**scatter_kwargs)
ax2.scatter(*rodney.li_dong_2016_load_cl(), label='Li & Dong (2016)',
**scatter_kwargs)
ax3.scatter(*rodney.li_dong_2016_load_cz(), label='Li & Dong (2016)',
**scatter_kwargs)
ax1.legend(frameon=False, prop=dict(size=10))
fig.tight_layout()
if args.save_figures:
figdir = maindir / 'figures'
figdir.mkdir(parents=True, exist_ok=True)
filepath = figdir / 'force_coefficients_compare_atol.png'
fig.savefig(filepath, dpi=300, bbox_inches='tight')
if args.show_figures:
pyplot.show()