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

这个项目还处于早期阶段，我希望及时使其更加全面:)