matplotlib_ai

matplotlib_ai

คุณมีความสัมพันธ์แบบรักและเกลียดกับ matplotlib หรือไม่? ฉันก็เช่นกัน! นั่นเป็นเหตุผลที่ฉันสร้างมินิโปรเจ็กต์ขึ้นมาซึ่งสามารถช่วยคุณสร้างกราฟข้อมูลโดยใช้ภาษาธรรมชาติได้ การพึ่งพาแพ็คเกจต้องใช้ openai และ matplotlib และใช้งานง่ายอย่างไม่น่าเชื่อ ด้วยการเรียก GPT API ของ OpenAI วิศวกรรมที่รวดเร็ว และใช้การเรียนรู้แบบไม่กี่ช็อต matplotlib_ai สามารถสร้างกราฟโดยไม่จำเป็นต้องให้คุณเขียนโค้ด matplotlib เพียงบรรทัดเดียว!

นำเข้า matplotlib_ai ผ่าน pip:

 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' เป็นเส้นประแล้วเรียกกราฟนี้ว่า "ekg ของฉันเมื่อฉันเห็นคุณ :)" สิ่งที่สมเหตุสมผลที่สุดคือการเขียนโค้ด 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()

โครงการนี้ยังอยู่ในช่วงเริ่มต้น ฉันหวังว่าจะทำให้ครอบคลุมมากขึ้นทันเวลา :)