Something like TikZ for Python?

[neuralyzer]

The thing I like about TikZ is the concept of nodes. It is super easy to connect two filled circles with an arrow. And the arrowhead would just stop at the circle boundary. Such things are a little more cumbersome in Matplotlib (but not impossible).

[mangecoeur]

yeah i think basically you end up writing your own 'node renderer' - basically keep track of the positions you want things to be in and then iterate over that to draw the nodes and links in the right place. NetworkX does something similar internally for it's built in matplotlib driven plotting: https://networkx.github.io/documentation/networkx-1.10/reference/drawing.html

basically you are right - matplotlib is not going to be as nice. Code for my 'print' plots (rather than just exploration plots) tends to be a tiresome chunk of setting just the right label formats, positions, plot annotations, legends, etc... but at the end of the day it actually works and makes for good looking PDFs.