Acabei de descobrir uma ótima maneira de criar um gráfico de contorno preenchido com Matplotlib cortado em uma região poligonal arbitrária. O método requer a chamadaset_clip_path(patch) em cada instância PathCollection no QuadContourSet retornado pelo Matplotlibcontourf() função. 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()

Estou confuso sobre um aspecto: se eu comentar a linha que representa o patch, nenhum dos recortes funciona e eu termino com um gráfico em branco. Presumo que, ao chamar o método set_clip_path com um patch no PathCollection, o patch deve ter sido adicionado aos eixos, mas não entendo o porquê. Configuraçãoedgecolor='none' para a criação de patches é uma boa solução, mas onde está a graça nisso?

Alguma ideia?