Início / coding / Perguntas / 79554664
Accepted
Paul
Asked: 2025-04-04 15:08:21 +0800 CST

aviso de barra de cores com pcolor e np.nan

  • 772

Tenho um array com valores np.nan que quero plotar usando pcolor. Em princípio, tudo funciona, mas recebo um aviso do qual não consigo me livrar. Usar plt.imshownão dá o aviso, mas preciso especificar as coordenadas x e y.

MatplotlibDeprecationWarning: Obter o array de um PolyQuadMesh retornará o array completo no futuro (descompactado). Para obter esse comportamento, defina o PolyQuadMesh com um array 2D .set_array(data2d).

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt

x = np.linspace(-2,2,100)
y = np.linspace(-2,2,100)

X, Y = np.meshgrid(x,y)
X[X**2+Y**2>4] = np.nan
Y[X**2+Y**2>4] = np.nan
Z = np.exp(-(X**2+Y**2))

plt.pcolor(Y,X,Z, cmap='viridis')

plt.colorbar()

Eu uso matplotlib v3.9.2 e numpy v1.26.4.

python

1 respostas

  1. Best Answer

    Este aviso é devido a uma mudança em como a pcolorlógica interna é estruturada ( changenote aqui ). Ele é acionado quando o código colorbar chama internamente get_arrayo objeto retornado por pcolor. Você pode silenciar o aviso repassando explicitamente seu Zarray para o set_arraymétodo:

    pc = plt.pcolor(Y,X,Z, cmap='viridis')
pc.set_array(Z)

plt.colorbar()

    Como alternativa, você pode atualizar seu Matplotlib para a versão 3.10+, já que essa descontinuação já expirou.

    • 1

relate perguntas