pandas gpt

pandas-gpt

Aprimore seu fluxo de trabalho de ciência de dados com ChatGPT.

---

pandas-gpt é uma biblioteca Python para fazer quase tudo com um DataFrame do pandas usando prompts ChatGPT.

Instalação

pip install pandas-gpt

Defina a variável de ambiente OPENAI_API_KEY como sua chave de API OpenAI ou use o seguinte snippet de código:

 import openai
openai . api_key = '' 

Exemplos

Exemplos de configuração e uso estão disponíveis neste notebook do Google Colab .

 import pandas as pd
import pandas_gpt

df = pd . DataFrame ( 'https://gist.githubusercontent.com/bluecoconut/9ce2135aafb5c6ab2dc1d60ac595646e/raw/c93c3500a1f7fae469cba716f09358cfddea6343/sales_demo_with_pii_and_all_states.csv' )

# Data transformation
df = df . ask ( 'drop purchases from Laurenchester, NY' )
df = df . ask ( 'add a new Category column with values "cheap", "regular", or "expensive"' )

# Queries
weekday = df . ask ( 'which day of the week had the largest number of orders?' )
top_10 = df . ask ( 'what are the top 10 most popular products, as a table' )

# Plotting
df . ask ( 'plot monthly and hourly sales' )
top_10 . ask ( 'horizontal bar plot with pastel colors' )

# Allow changes to original dataset
df . ask ( 'do something interesting' , mutable = True )

# Show source code before running
df . ask ( 'convert prices from USD to GBP' , verbose = True )

Outros anfitriões

Se você quiser usar um host de API diferente, como o serviço Azure OpenAI:

 import openai
openai . api_type = 'azure'
openai . api_base = ''
openai . api_version = ''
openai . api_key = ''

import pandas_gpt
# pandas_gpt.model = '' # Default is 'gpt-3.5-turbo'
pandas_gpt . completion_config = {
  'engine' : '' ,
  # 'deployment_id': '',
}

Alternativas

• GitHub Copilot: conclusão de código de uso geral (assinatura paga)
• Sketch: resumo de dados e sugestões de código com tecnologia de IA (funciona sem uma chave de API)

Isenção de responsabilidade

Observe que as limitações do ChatGPT também se aplicam a esta biblioteca. Eu recomendaria usar pandas-gpt em um ambiente sandbox, como Google Colab, Kaggle ou GitPod.