matplotlib_ai
Apakah Anda juga memiliki hubungan cinta-benci dengan matplotlib? Aku juga! Itu sebabnya saya membuat proyek mini ini yang dapat membantu Anda membuat grafik data menggunakan bahasa alami. Dependensi paket memerlukan openai
dan matplotlib
, dan sangat mudah digunakan. Memanggil API GPT OpenAI, rekayasa cepat, dan menggunakan pembelajaran beberapa langkah, matplotlib_ai
mampu menghasilkan grafik tanpa mengharuskan Anda menulis satu baris kode matplotlib
!
Impor matplotlib_ai
melalui pip:
pip install matplotlib_ai
Katakanlah kita mempunyai data
kamus dengan 4 kurva berlabel 'a'
, 'b'
, 'c'
, dan 'd'
:
import numpy as np
data = { 'a' : [...], # some curve
'b' : [...], # some curve
'c' : [...], # some curve
'd' : [...], # some curve}
Jika kita ingin membuat grafik setiap kurva dan membuat kurva 'a'
putus-putus dan menyebut grafik ini "ekg saya ketika saya melihat Anda :)", hal yang paling masuk akal adalah menulis kode matplotlib
seperti:
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 ()
Namun, dengan matplotlib_ai
semudah:
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 )
Kemudian, mpl_ai
akan menghasilkan:
Untuk melihat kode yang dihasilkan oleh GPT, cukup cetak seperti ini:
>>> 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()
Proyek ini masih dalam tahap awal, saya berharap dapat membuatnya lebih komprehensif pada waktunya :)