Python Pandas是一個為Python編程提供數據操作和分析功能的開放原始碼工具包。過去它已成為數據科學家和分析師的必備工具。現在我們可以搭配使用 ChatGPT 讓 Pandas提供了一種更有效的方法來管理結構化數據(Series和DataFrame)。

Pandas AI 是一個 Python 庫，它將生成人工智能功能集成到 Pandas 中，讓我們可以透過類似 ChatGPT 的對話做資料處理

在人工智慧領域，Pandas 經常用於機器學習和深度學習過程的預處理步驟。Pandas通過提供數據清理、重塑、合併和聚合，可以將原始數據集轉換為結構化的、隨時可用的2維表格，並將其輸入人工智慧演算法。

項目地址：https://github.com/gventuri/pandas-ai

使用 pip 安裝 Pandas AI

pip install pandasai

使用 OpenAI 導入 PandasAI

在下一步中，我們將導入之前安裝的 pandasai 庫，然後導入 LLM（大型語言模型）功能。截至 2023 年 5 月，pandasai 僅支持 OpenAI 模型，我們將使用它來理解數據。

import pandas as pd
from pandasai import PandasAI

# Sample DataFrame
df = pd.DataFrame({
    "country": ["United States", "United Kingdom", "France", "Germany", "Italy", "Spain", "Canada", "Australia", "Japan", "China"],
    "gdp": [19294482071552, 2891615567872, 2411255037952, 3435817336832, 1745433788416, 1181205135360, 1607402389504, 1490967855104, 4380756541440, 14631844184064],
    "happiness_index": [6.94, 7.16, 6.66, 7.07, 6.38, 6.4, 7.23, 7.22, 5.87, 5.12]
})

# Instantiate a LLM
from pandasai.llm.openai import OpenAI
llm = OpenAI(api_token="your_API_key")

pandas_ai = PandasAI(llm)
pandas_ai.run(df, prompt='Which are the 5 happiest countries?')

6            Canada
7         Australia
1    United Kingdom
3           Germany
0     United States
Name: country, dtype: object

要使用 OpenAI API，您必須輸入自己的 OpenAI API Key。

因為pandas的特性，我們不僅僅可以處理csv文件，我們還可以連接關係型的資料庫，例如pgsql：

# creating the uri and connecting to database
 pg_conn = "postgresql://YOUR URI HERE"
 
 #Query sql database 
 query = """
 SELECT *
 FROM table_name
 """
 
 #Create dataframe named df
 df = pd.read_sql(query,pg_conn)

然後像上面程式碼一樣，我們可以直接與它進行對話了：

# Using pandas-ai!
 pandas_ai = PandasAI(llm)
 pandas_ai.run(df, prompt='Place your prompt here)

當然，你也可以讓 PandasAI 進行更複雜的查詢。例如，可以要求 PandasAI 求出 2 個最不幸福國家的 GDP 總和：

pandas_ai.run(df, prompt='What is the sum of the GDPs of the 2 unhappiest countries?')

上面的程式碼將返回以下內容：

19012600725504

也可以請 PandasAI 產生圖表：

pandas_ai.run(
    df,
    "Plot the histogram of countries showing for each the gpd, using different colors for each bar",
)

最後

ChatGPT、Pandas 是強大的工具，當它們結合在一起時，可以改變我們與數據交互和分析的方式。ChatGPT 憑借其先進的自然語言處理能力，可以更直觀地與數據進行類似人類的交互。而 PandasAI 可以增強Pandas數據分析體驗。通過將複雜的數據操作任務轉換為簡單的自然語言查詢，PandasAI 使用戶更容易從數據中提取有價值的見解，而無需像過去一樣編寫大量的 Python 程式碼。

這對於那些還不熟悉 Python 或 pandas 操作/轉換的人來說是一種做資料處理的新方法。我們不需要為你想要執行的任務編程，而是只是與AI代理交談，明確的告訴它想要的結果，PandaAI 替你轉換為電腦可解釋的程式碼，並幫助您做資料處理。

Share this post