Geração de títulos de estímulo

[

Tutorial de consulta pandas em lista

Neste tutorial, aprenderemos como usar a função query do pandas para realizar consultas em listas. A função query é uma maneira eficiente de filtrar dados em um DataFrame com base em uma condição específica.

Antes de começarmos, vamos garantir que as bibliotecas pandas e numpy estejam instaladas no seu ambiente Python. Caso contrário, você pode instalá-las usando o comando pip install pandas numpy.

Importando bibliotecas necessárias

import pandas as pd
import numpy as np

Criando um DataFrame de exemplo

Para este tutorial, vamos criar um DataFrame de exemplo contendo informações sobre funcionários de uma empresa.

data = {'Nome': ['João', 'Maria', 'Pedro', 'Ana'],
        'Idade': [28, 32, 24, 27],
        'Departamento': ['RH', 'Vendas', 'TI', 'Marketing']}
df = pd.DataFrame(data)

Aqui está uma visualização do DataFrame:

   Nome  Idade Departamento
0  João     28           RH
1  Maria    32        Vendas
2  Pedro    24           TI
3  Ana      27     Marketing

Verificando se um elemento está em uma lista (pandas)

Para verificar se um elemento está presente em uma lista usando o pandas, podemos usar o método isin em conjunto com a função any.

# Verificando se o nome "João" está presente na coluna "Nome"
nome_presente = df['Nome'].isin(['João']).any()
print(nome_presente)

# Verificando se o nome "Lucas" está presente na coluna "Nome"
nome_presente = df['Nome'].isin(['Lucas']).any()
print(nome_presente)

Saída:

True
False

Se um elemento estiver presente na lista, o resultado será True, caso contrário, será False.

Verificando se uma string está em uma lista (pandas)

Para verificar se uma string está presente em uma lista usando o pandas, podemos usar o método str.contains em conjunto com a função any.

# Verificando se a string "RH" está presente na coluna "Departamento"
string_presente = df['Departamento'].str.contains('RH').any()
print(string_presente)

# Verificando se a string "Financeiro" está presente na coluna "Departamento"
string_presente = df['Departamento'].str.contains('Financeiro').any()
print(string_presente)

Saída:

True
False

Se a string estiver presente na lista, o resultado será True, caso contrário, será False.

Obtendo linhas que estão em uma lista (pandas)

Para obter as linhas que estão presentes em uma lista usando o pandas, podemos usar o método query em conjunto com a função isin.

# Obtendo as linhas com nomes "João" e "Pedro"
linhas = df.query("Nome.isin(['João', 'Pedro'])")
print(linhas)

Saída:

   Nome  Idade Departamento
0  João     28           RH
2  Pedro    24           TI

O método query permite que você escreva uma expressão booleana para filtrar os dados do DataFrame. Usando o método isin, podemos especificar uma lista de valores a serem procurados.

Obtendo colunas pandas em uma lista

Para obter as colunas do pandas em uma lista, podemos usar o método columns do DataFrame.

# Obtendo todas as colunas em uma lista
colunas = df.columns.tolist()
print(colunas)

Saída:

['Nome', 'Idade', 'Departamento']

O método columns retorna um objeto Index contendo os nomes das colunas do DataFrame. Usando o método tolist(), podemos convertê-lo em uma lista simples.

Conclusão

Neste tutorial, aprendemos como usar a função query do pandas para realizar consultas em listas. Também vimos como verificar se um elemento ou string está presente em uma lista usando os métodos isin e str.contains, e como obter linhas e colunas que estão em uma lista. Espero que este tutorial tenha sido útil para você aprender mais sobre consultas em listas com o pandas.