Criando várias subtramas usando `plt.subplots`#

pyplot.subplotscria uma figura e uma grade de subtramas com uma única chamada, enquanto fornece controle razoável sobre como as parcelas individuais são criadas. Para casos de uso mais avançados, você pode usar GridSpecpara um layout de subtrama mais geral ou Figure.add_subplotpara adicionar subtramas em locais arbitrários dentro da figura.

import matplotlib.pyplot as plt
import numpy as np

# Some example data to display
x = np.linspace(0, 2 * np.pi, 400)
y = np.sin(x ** 2)

Uma figura com apenas uma subtrama #

subplots()sem argumentos retorna um Figuree um único Axes.

Esta é realmente a maneira mais simples e recomendada de criar uma única Figura e Eixos.

fig, ax = plt.subplots()
ax.plot(x, y)
ax.set_title('A single plot')

Text(0.5, 1.0, 'A single plot')

Empilhando subparcelas em uma direção #

Os dois primeiros argumentos opcionais de pyplot.subplotsdefinem o número de linhas e colunas da grade da subtrama.

Ao empilhar em apenas uma direção, o retornado axsé um array numpy 1D contendo a lista de eixos criados.

fig, axs = plt.subplots(2)
fig.suptitle('Vertically stacked subplots')
axs[0].plot(x, y)
axs[1].plot(x, -y)

[<matplotlib.lines.Line2D object at 0x7f2d00efd510>]

Se você estiver criando apenas alguns eixos, é útil descompactá-los imediatamente em variáveis dedicadas para cada eixo. Dessa forma, podemos usar ax1em vez do mais detalhado axs[0].

fig, (ax1, ax2) = plt.subplots(2)
fig.suptitle('Vertically stacked subplots')
ax1.plot(x, y)
ax2.plot(x, -y)

[<matplotlib.lines.Line2D object at 0x7f2d00a95b70>]

Para obter subtramas lado a lado, passe parâmetros para uma linha e duas colunas.1, 2

fig, (ax1, ax2) = plt.subplots(1, 2)
fig.suptitle('Horizontally stacked subplots')
ax1.plot(x, y)
ax2.plot(x, -y)

[<matplotlib.lines.Line2D object at 0x7f2cfb43d330>]

Empilhando subtramas em duas direções #

Ao empilhar em duas direções, o retornado axsé um array NumPy 2D.

Se você tiver que definir parâmetros para cada subplot, é útil iterar todos os subplots em uma grade 2D usando .for ax in axs.flat:

fig, axs = plt.subplots(2, 2)
axs[0, 0].plot(x, y)
axs[0, 0].set_title('Axis [0, 0]')
axs[0, 1].plot(x, y, 'tab:orange')
axs[0, 1].set_title('Axis [0, 1]')
axs[1, 0].plot(x, -y, 'tab:green')
axs[1, 0].set_title('Axis [1, 0]')
axs[1, 1].plot(x, -y, 'tab:red')
axs[1, 1].set_title('Axis [1, 1]')

for ax in axs.flat:
    ax.set(xlabel='x-label', ylabel='y-label')

# Hide x labels and tick labels for top plots and y ticks for right plots.
for ax in axs.flat:
    ax.label_outer()

Eixo [0, 0], Eixo [0, 1], Eixo [1, 0], Eixo [1, 1]

Você pode usar a descompactação de tupla também em 2D para atribuir todas as subtramas a variáveis dedicadas:

fig, ((ax1, ax2), (ax3, ax4)) = plt.subplots(2, 2)
fig.suptitle('Sharing x per column, y per row')
ax1.plot(x, y)
ax2.plot(x, y**2, 'tab:orange')
ax3.plot(x, -y, 'tab:green')
ax4.plot(x, -y**2, 'tab:red')

for ax in fig.get_axes():
    ax.label_outer()

Compartilhando x por coluna, y por linha

Eixos polares #

O parâmetro subplot_kw de pyplot.subplotscontrola as propriedades da subtrama (consulte também Figure.add_subplot). Em particular, isso pode ser usado para criar uma grade de eixos polares.

fig, (ax1, ax2) = plt.subplots(1, 2, subplot_kw=dict(projection='polar'))
ax1.plot(x, y)
ax2.plot(x, y ** 2)

plt.show()

Tempo total de execução do script: (0 minutos 7,774 segundos)

Galeria gerada por Sphinx-Gallery

Criando várias subtramas usando plt.subplots#