Plot of cos(tan(rica)) where rica in [0, pi]

costa_rica.py

from matplotlib.pyplot import subplots
from numpy import cos, tan, linspace, pi

fig, ax = subplots(1,1)
rica = linspace(0,pi,500)
costa_rica = cos(tan(rica))
#
ax.plot(
    rica,
    costa_rica,
    "k"
)

bs = [-1, -2/3,-.25,.25,2/3]
colors = ["#002b7f","#ffffff", "#ce1126","#ffffff", "#002b7f"]

for c, b in zip(colors, bs):
    ax.fill_between(rica, b, costa_rica, color=c, where=costa_rica>b)

ax.set_title(r"$f(rica)=\cos(\tan(rica))$", fontsize=20);
ax.set_xlabel(r"$rica$", fontsize=16)
ax.set_ylabel(r"$f(rica)$", fontsize=16)
fig.set_size_inches(12,7)