Alguém sabe como desenhar uma borda ao redor de uma subtrama individual dentro de uma figura no matplotlib? Estou usando o pyplot.

por exemplo:

import matplotlib.pyplot as plt
f = plt.figure()
ax1 = f.add_subplot(211)
ax2 = f.add_subplot(212)
# ax1.set_edgecolor('black')

..mas os objetos Axes não têm 'edgecolor', e não consigo encontrar uma maneira de delinear o gráfico a partir do nível da figura.

Na verdade, estou envolvendo o código mpl e adicionando uma interface do wx com controles que gostaria de ter contexto, dependendo de qual subtrama está selecionado. Ou seja, o usuário clica na subtrama na figura da tela - a subtrama é 'selecionada' (tem um contorno desenhado em torno dela, idealmente dente de serra) - a GUI é atualizada para apresentar controles para modificar essa subtrama específica.