autoplotlib

Gere rapidamente gráficos em Python simplesmente descrevendo -os através do texto. Por exemplo, basta usar o prompt "Criar um gráfico de dispersão com rótulos de nomes" em vez de pesquisar infinitamente nos documentos e StackOverflow. Dado esse aviso, a biblioteca python autoplotlib usa a API do OpenAI para gerar automaticamente o código e plotagem correspondentes - economizando tempo e esforço.

Aviso que este pacote é experimental. O pacote pode executar a saída de código a partir de modelos de idiomas. Embora improvável, sempre verifique e confirme nenhum código malicioso foi gerado pelo modelo antes da execução. Use por sua conta e risco.

Instalação

1. Instale o pacote Python:

    pip install autoplotlib
   

2. Obtenha uma chave de API do OpenAI, inscrevendo -se aqui se você ainda não tiver um.

Uso

A seguir, é apresentado um exemplo de plotagem de um quadro de dados de pandas usando a biblioteca autoplotlib . A função plot() leva um prompt que descreve o gráfico e os dados para plotar. A função retorna o gráfico, o código gerado e a resposta completa da API do OpenAI.

 # set the OpenAI API key
import os
os . environ [ "OPENAI_API_KEY" ] = "<YOUR_API_KEY>"

import autoplotlib as aplt
import pandas as pd

data = pd . DataFrame (
    [[ 29 , 177 ], [ 33 , 186 ], [ 48 , 161 ], [ 53 , 173 ], [ 67 , 152 ]],
    index = [ "Alice" , "Bob" , "Charlie" , "Dave" , "Eve" ],
    columns = [ "age" , "height" ],
)

figure_description = """
Plot the data as scatterplot between height and age.
Add the names as labels next to the data points.
Ensure the labels don't overlap.
Mark people taller than 170 with a star instead of a point.
"""

code , fig , llm_response = aplt . plot ( figure_description , data = data )

Contribuindo

Consulte o Guia de Desenvolvimento para obter informações sobre como contribuir.

Recursos futuros em potencial

• Adicionar sandboxing da execução do código (veja aqui)

Licença

MIT Licença