matplotlib_ai

matplotlib_ai

Você também tem uma relação de amor e ódio com o matplotlib? Eu também! É por isso que criei este miniprojeto que pode ajudá-lo a representar graficamente seus dados usando linguagem natural. As dependências do pacote requerem openai e matplotlib e são incrivelmente fáceis de usar. Chamando a API GPT da OpenAI, engenharia imediata e usando aprendizado rápido, matplotlib_ai é capaz de gerar gráficos sem exigir que você escreva uma única linha de código matplotlib !

Importe matplotlib_ai via pip:

 pip install matplotlib_ai

Digamos que temos data de um dicionário com 4 curvas rotuladas 'a' , 'b' , 'c' e 'd' :

 import numpy as np
data = { 'a' : [...], # some curve
        'b' : [...], # some curve
	'c' : [...], # some curve
	'd' : [...], # some curve}

Se quiséssemos representar graficamente cada curva e fazer a curva 'a' tracejada e chamar esse gráfico de "meu ekg quando te vejo :)", a coisa mais sensata seria escrever o código matplotlib como tal:

 import matplotlib . pyplot as plt
plt . plot ( data [ 'a' ], linestyle = 'dashed' , label = 'a' )
plt . plot ( data [ 'b' ], label = 'b' )
plt . plot ( data [ 'c' ], label = 'c' )
plt . plot ( data [ 'd' ], label = 'd' )
plt . title ( 'my ekg when i see you :)' )
plt . legend ()
plt . show ()

No entanto, com matplotlib_ai é tão fácil quanto:

 from matplotlib_ai . matplotlib_ai import matplotlib_ai
mpl_ai = matplotlib_ai ( "YOUR-OPENAI-API-KEY" )
prompt = "graph a curve for each item in data and title the graph 'my ekg when i see you :)'. " + 
	 "Make curve 'a' in data a dashed line."
code = mpl_ai ( prompt )

Então, mpl_ai geraria:

Para ver o código gerado pelo GPT, basta imprimi-lo assim:

 >>> print(code)		# the code generated by GPT
import matplotlib.pyplot as plt
for key, value in data.items():
    if key == 'a':
        plt.plot(value, linestyle='dashed', label=key)
    else:
        plt.plot(value, label=key)
plt.title('my ekg when i see you :)')
plt.legend()
plt.show()

Este projeto está em seus estágios iniciais, espero torná-lo mais abrangente com o tempo :)