Colorir por Valor de Y

Beginner

This tutorial is from open-source community. Access the source code

Introdução

Matplotlib é uma biblioteca de visualização de dados em Python. É uma ferramenta poderosa para criar uma ampla gama de gráficos e diagramas. Uma das funcionalidades do Matplotlib é a capacidade de plotar linhas com cores diferentes com base no valor de y. Este laboratório demonstrará como usar arrays mascarados para plotar uma linha com cores diferentes por valor de y.

Dicas para a VM

Após a inicialização da VM, clique no canto superior esquerdo para mudar para a aba Notebook e acessar o Jupyter Notebook para praticar.

Às vezes, pode ser necessário aguardar alguns segundos para que o Jupyter Notebook termine de carregar. A validação das operações não pode ser automatizada devido a limitações no Jupyter Notebook.

Se você enfrentar problemas durante o aprendizado, sinta-se à vontade para perguntar ao Labby. Forneça feedback após a sessão, e resolveremos o problema prontamente para você.

Importar as Bibliotecas Necessárias

Nesta etapa, importaremos as bibliotecas necessárias para este laboratório.

import matplotlib.pyplot as plt
import numpy as np

Criar Dados

Nesta etapa, criaremos os dados para nosso gráfico. Criaremos um array de valores para t e um array de valores para s.

t = np.arange(0.0, 2.0, 0.01)
s = np.sin(2 * np.pi * t)

Criar Arrays Mascarados

Nesta etapa, criaremos três arrays mascarados: um para valores maiores que um determinado limite (threshold), um para valores menores que um determinado limite e um para valores entre dois limites.

upper = 0.77
lower = -0.77

supper = np.ma.masked_where(s < upper, s)
slower = np.ma.masked_where(s > lower, s)
smiddle = np.ma.masked_where((s < lower) | (s > upper), s)

Criar o Gráfico

Nesta etapa, criaremos o gráfico usando os arrays mascarados criados na etapa anterior. Plotaremos cada array mascarado separadamente e usaremos cores diferentes para cada um.

fig, ax = plt.subplots()
ax.plot(t, smiddle, t, slower, t, supper)
plt.show()

Resumo

Neste laboratório, aprendemos como plotar linhas com cores diferentes com base no valor de y, usando arrays mascarados em Matplotlib. Essa técnica pode ser útil ao visualizar dados com regiões de interesse distintas que exigem cores diferentes para maior clareza.