Como adicionar título a subquadrantes em Matplotlib

Suraj Joshi 15 fevereiro 2024
  1. Método Set_title() para adicionar título ao subplot em Matplotlib
  2. title.set_text() Método para definir o título de subquadros em Matplotlib
  3. plt.gca().set_title() / plt.gca.title.set_text() para definir título para subquadrantes em Matplotlib
Como adicionar título a subquadrantes em Matplotlib

Utilizamos métodos set_title(label) e title.set_text(label) para adicionar títulos a subquadrantes em Matplotlib.

Método Set_title() para adicionar título ao subplot em Matplotlib

Utilizamos matplotlib.axes._axes.Axes.set_title(label) método para definir o título (string label) para o subplot atual Axes.

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 100)
y1 = np.sin(x)
y2 = np.cos(x)
y3 = 1 / (1 + np.exp(-x))
y4 = np.exp(x)

fig, ax = plt.subplots(2, 2)

ax[0, 0].plot(x, y1)
ax[0, 1].plot(x, y2)
ax[1, 0].plot(x, y3)
ax[1, 1].plot(x, y4)

ax[0, 0].set_title("Sine function")
ax[0, 1].set_title("Cosine function")
ax[1, 0].set_title("Sigmoid function")
ax[1, 1].set_title("Exponential function")

fig.tight_layout()
plt.show()

Resultado:

set_title para adicionar título ao subplot em Matplotlib

Se quisermos passar por cima de alguns subquadros e mostrá-los um de cada vez junto com os títulos, podemos usar o seguinte código mais curto:

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 100)
y = [0, 0, 0, 0]
y[0] = np.sin(x)
y[1] = np.cos(x)
y[2] = 1 / (1 + np.exp(-x))
y[3] = np.exp(x)

figure, ax = plt.subplots(2, 2)
i = 0
for a in range(len(ax)):
    for b in range(len(ax[a])):
        ax[a, b].plot(x, y[i])
        subplot_title = "Subplot" + str(i)
        ax[a, b].set_title(subplot_title)
        i = i + 1

figure.tight_layout()
plt.show()

Resultado:

set_title para adicionar título ao subplot em Matplotlib - método loop

title.set_text() Método para definir o título de subquadros em Matplotlib

Também podemos adicionar título aos subquadros em Matplotlib utilizando o método title.set_text(), de forma similar ao método set_title().

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 100)
y1 = np.sin(x)
y2 = np.cos(x)
y3 = 1 / (1 + np.exp(-x))
y4 = np.exp(x)

fig, ax = plt.subplots(2, 2)

ax[0, 0].plot(x, y1)
ax[0, 1].plot(x, y2)
ax[1, 0].plot(x, y3)
ax[1, 1].plot(x, y4)

ax[0, 0].title.set_text("Sine function")
ax[0, 1].title.set_text("Cosine function")
ax[1, 0].title.set_text("Sigmoid function")
ax[1, 1].title.set_text("Exponential function")

fig.tight_layout()
plt.show()

Resultado:

title.set_text Método para adicionar título ao subplot em Matplotlib

plt.gca().set_title() / plt.gca.title.set_text() para definir título para subquadrantes em Matplotlib

Se você utilizar o estilo Matlab na trama interativa, então você poderia utilizar plt.gca() para obter a referência dos eixos atuais do subplot e combinar set_title() ou title.set_text() método para definir o título dos subquadros em Matplotlib.

import matplotlib.pyplot as plt

plt.subplots(2, 2)
x = [1, 2, 3]
y = [2, 4, 6]


for i in range(4):
    plt.subplot(2, 2, i + 1)
    plt.plot(x, y)
    plt.gca().set_title("Title-" + str(i))

plt.show()

plt.tight_layout()

ou,

import matplotlib.pyplot as plt

plt.subplots(2, 2)
x = [1, 2, 3]
y = [2, 4, 6]


for i in range(4):
    plt.subplot(2, 2, i + 1)
    plt.plot(x, y)
    plt.gca().title.set_text("Title-" + str(i))

plt.show()

plt.tight_layout()
Suraj Joshi avatar Suraj Joshi avatar

Suraj Joshi is a backend software engineer at Matrice.ai.

LinkedIn

Artigo relacionado - Matplotlib Subplot