matplotlib_ai

matplotlib_ai

你是不是也對 matplotlib 又愛又恨呢？我也是！這就是為什麼我創建了這個迷你項目，它可以幫助您使用自然語言繪製資料圖表。軟體包依賴項需要openai和matplotlib ，而且它非常容易使用。呼叫 OpenAI 的 GPT API，快速工程，並使用少樣本學習， matplotlib_ai能夠產生圖形，而無需您編寫一行matplotlib程式碼！

透過 pip 導入matplotlib_ai ：

 pip install matplotlib_ai

假設我們有一個字典data ，其中有 4 條曲線，分別標示為'a' 、 'b' 、 'c'和'd' ：

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

如果我們想繪製每條曲線並將曲線'a'繪製為虛線，並將該圖稱為“我看到你時的心電圖:)”，最明智的做法是編寫matplotlib程式碼，如下所示：

 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 ()

然而，使用matplotlib_ai就很簡單：

 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 )

然後， mpl_ai將產生：

要查看 GPT 產生的程式碼，只需像這樣列印：

 >>> 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()

這個專案還處於早期階段，我希望及時使其更加全面:)