matplotlib_ai

matplotlib_ai

matplotlib과도 애증 관계가 있습니까? 나도 마찬가지야! 이것이 바로 제가 자연어를 사용하여 데이터를 그래프로 표시하는 데 도움이 되는 이 미니 프로젝트를 만든 이유입니다. 패키지 종속성에는 openai 및 matplotlib 필요하며 믿을 수 없을 정도로 사용하기 쉽습니다. OpenAI의 GPT API 호출, 프롬프트 엔지니어링 및 몇 번의 학습을 사용하는 matplotlib_ai matplotlib 코드 한 줄도 작성하지 않고도 그래프를 생성할 수 있습니다!

pip를 통해 matplotlib_ai 가져옵니다.

 pip install matplotlib_ai

'a' , 'b' , 'c' 및 'd' 라벨이 붙은 4개의 곡선이 있는 사전 data 있다고 가정해 보겠습니다.

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

이 프로젝트는 초기 단계이므로 시간이 지나면 더욱 포괄적으로 진행되기를 바랍니다. :)