Acabo de descubrir una buena manera de crear un diagrama de contorno lleno de Matplotlib recortado a una región poligonal arbitraria. El método requiere llamarset_clip_path(patch) en cada instancia de PathCollection en QuadContourSet devuelta por Matplotlib'scontourf() función. MWE:

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.patches as mpatches
import matplotlib.path as mpath

# some arbitrary data to plot
xx, yy = np.meshgrid(np.linspace(-5, 5, 20), np.linspace(-10, 10, 20), copy=False)
zz = np.sqrt(xx ** 2 + yy ** 2)

poly_verts = [
    (0, 0),
    (-4, 7),
    (-4, -7),
    (4, -7),
    (4, 7),
    (0, 0)
]
poly_codes = [mpath.Path.MOVETO] + (len(poly_verts) - 2) * [mpath.Path.LINETO] +
mpath.Path.CLOSEPOLY]

# create a Path from the polygon vertices
path = mpath.Path(poly_verts, poly_codes)

# create a Patch from the path
patch = mpatches.PathPatch(path, facecolor='none', edgecolor='k')

plt.figure()
ax = plt.gca()
cont = plt.contourf(xx, yy, zz, 50)

# add the patch to the axes
ax.add_patch(patch)  ## TRY COMMENTING THIS OUT
for col in cont.collections:
    col.set_clip_path(patch)

plt.show()

Estoy confundido acerca de un aspecto: si comento la línea que traza el parche, entonces ninguno de los recortes funciona y termino con un diagrama en blanco. Supongo que cuando se llama al método set_clip_path con un parche en PathCollection, el parche debe haberse agregado a los ejes, pero no entiendo por qué. Ajusteedgecolor='none' para la creación de parches es una buena solución, pero ¿dónde está la diversión en eso?

¿Alguna idea?