linestyles.py

# -----------------------------------------------------------------------------
# Matplotlib cheat sheet
# Released under the BSD License
# -----------------------------------------------------------------------------
importpathlib

importnumpyasnp
importmatplotlib.pyplotasplt


ROOT_DIR=pathlib.Path(__file__).parent.parent

fig=plt.figure(figsize=(4.25, 2*.55))
ax=fig.add_axes([0, 0, 1, 1], xlim=[0.75, 10.25], ylim=[0.5, 2.5], frameon=False,
xticks=[], yticks=[])
y=2


defsplit(n_segment):
width=9
segment_width=0.75*(width/n_segment)
segment_pad= (width-n_segment*segment_width)/(n_segment-1)
X0=1+np.arange(n_segment)*(segment_width+segment_pad)
X1=X0+segment_width
returnX0, X1


# Line style
# ----------------------------------------------------------------------------
X0, X1=split(5)
styles="-", ":", "--", "-.", (0, (0.01, 2))

forx0, x1, styleinzip(X0, X1, styles):
ax.plot([x0, x1], [y, y], color="C1", linestyle=style,
solid_capstyle="round", dash_capstyle="round", linewidth=3)
ifisinstance(style, str): text='"%s"'%style
else: text='%s'%str(style)
text=text.replace(' ', '')
ax.text((x0+x1)/2, y-0.2, text,
size=8, ha="center", va="top", family="Source Code Pro")
ax.text(X0[0]-0.25, y+0.2, "linestyle or ls", family="Source Code Pro",
size=14, ha="left", va="baseline")
y-=1

# Dash capstyle
# ----------------------------------------------------------------------------
X0, X1=split(3)
styles="butt", "round", "projecting"
forx0, x1, styleinzip(X0, X1, styles):
ax.plot([x0, x1], [y, y], color="C1", dash_capstyle="projecting",
linewidth=7, linestyle="--", alpha=.25)
ax.plot([x0, x1], [y, y], color="C1", linewidth=7,
linestyle="--", dash_capstyle=style)
ax.text((x0+x1)/2, y-0.2, '"%s"'%style, family="Source Code Pro",
size=10, ha="center", va="top")
ax.text(X0[0]-0.25, y+0.2, "capstyle or dash_capstyle", family="Source Code Pro",
size=14, ha="left", va="baseline")


fig.savefig(ROOT_DIR/"figures/linestyles.pdf", dpi=200)
# plt.show()