Por que muitos exemplos usam `fig, ax = plt.subplots ()` em Matplotlib / pyplot / python

Estou aprendendo a usar matplotlibestudando exemplos, e muitos exemplos parecem incluir uma linha como a seguinte antes de criar um único gráfico ...

fig, ax = plt.subplots()

Aqui estão alguns exemplos...

• Modificar texto do rótulo de marca
• http://matplotlib.org/examples/pylab_examples/boxplot_demo2.html

Vejo essa função usada muito, mesmo que o exemplo esteja apenas tentando criar um único gráfico. Existe alguma outra vantagem? A demonstração oficial de subplots()também é usada f, ax = subplotsna criação de um único gráfico, e só faz referência a machado depois disso. Este é o código que eles usam.

# Just a figure and one subplot
f, ax = plt.subplots()
ax.plot(x, y)
ax.set_title('Simple plot')

plt.subplots()é uma função que retorna uma tupla contendo uma figura e eixos de objetos. Portanto, ao usar fig, ax = plt.subplots()você descompacta essa tupla nas variáveis fige ax. Ter figé útil se você deseja alterar os atributos no nível da figura ou salvar a figura como um arquivo de imagem posteriormente (por exemplo, com fig.savefig('yourfilename.png')). Você certamente não precisa usar o objeto de figura retornado, mas muitas pessoas o usam mais tarde, por isso é comum ver. Além disso, todos os objetos de eixos (os objetos que possuem métodos de plotagem) têm um objeto de figura pai de qualquer maneira, portanto:

fig, ax = plt.subplots()

é mais conciso do que isso:

fig = plt.figure()
ax = fig.add_subplot(111)

Apenas um suplemento aqui.

A seguinte pergunta é: e se eu quiser mais subtramas na figura?

Como mencionado no Doc, podemos usar fig = plt.subplots(nrows=2, ncols=2)para definir um grupo de subparcelas com grade (2,2) em um objeto de figura.

Então, como sabemos, o fig, ax = plt.subplots()retorna uma tupla, vamos tentar fig, ax1, ax2, ax3, ax4 = plt.subplots(nrows=2, ncols=2)primeiro.

ValueError: not enough values to unpack (expected 4, got 2)

Isso gera um erro, mas não se preocupe, porque agora vemos que plt.subplots()realmente retorna uma tupla com dois elementos. O primeiro deve ser um objeto de figura e o outro deve ser um grupo de objetos de subparcelas.

Então, vamos tentar novamente:

fig, [[ax1, ax2], [ax3, ax4]] = plt.subplots(nrows=2, ncols=2)

e verifique o tipo:

type(fig) #<class 'matplotlib.figure.Figure'>
type(ax1) #<class 'matplotlib.axes._subplots.AxesSubplot'>

Obviamente, se você usar parâmetros como (nrows = 1, ncols = 4), o formato deverá ser:

fig, [ax1, ax2, ax3, ax4] = plt.subplots(nrows=1, ncols=4)

Portanto, lembre-se de manter a construção da lista igual à grade de subparcelas que definimos na figura.

Espero que isso seja útil para você.

Como complemento à pergunta e às respostas acima, também há uma diferença importante entre plt.subplots()e plt.subplot(), observe a falta 's'no final.

Pode-se usar plt.subplots()para fazer todas as suas subparcelas de uma só vez e retornar a figura e os eixos (plural de eixo) das subparcelas como uma tupla. Uma figura pode ser entendida como uma tela em que você pinta seu esboço.

# create a subplot with 2 rows and 1 columns
fig, ax = plt.subplots(2,1)

Visto que você pode usar plt.subplot()se desejar adicionar as subparcelas separadamente. Retorna apenas o eixo de uma subtrama.

fig = plt.figure() # create the canvas for plotting
ax1 = plt.subplot(2,1,1) 
# (2,1,1) indicates total number of rows, columns, and figure number respectively
ax2 = plt.subplot(2,1,2)

No entanto, plt.subplots()é preferido porque oferece opções mais fáceis de personalizar diretamente toda a sua figura

# for example, sharing x-axis, y-axis for all subplots can be specified at once
fig, ax = plt.subplots(2,2, sharex=True, sharey=True)

enquanto que, com plt.subplot(), será necessário especificar individualmente para cada eixo que pode se tornar complicado.

Além das respostas acima, você pode verificar o tipo de objeto type(plt.subplots())que retorna uma tupla, por outro lado, type(plt.subplot())retornos matplotlib.axes._subplots.AxesSubplotque não podem ser descompactados.