Pandas AI為 Pandas 庫加入 AI 功能

Pandas AI 是一個 Python 庫，它為流行的 Python 數(shù)據(jù)分析和操作工具 Pandas 庫添加了生成人工智能功能，旨在與 Pandas 結(jié)合使用，而不是它的替代品。

Demo

在瀏覽器中試用 PandasAI：

安裝

pip install pandasai

用法

PandasAI 旨在與 Pandas 結(jié)合使用。它使 Pandas 具有對話性，允許以 Pandas DataFrames 的形式提出有關(guān)數(shù)據(jù)的問題并獲得答案。

例如，您可以要求 PandasAI 查找 DataFrame 中列值大于 5 的所有行，它將返回僅包含這些行的 DataFrame：

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": [21400000, 2940000, 2830000, 3870000, 2160000, 1350000, 1780000, 1320000, 516000, 14000000],
    "happiness_index": [7.3, 7.2, 6.5, 7.0, 6.0, 6.3, 7.3, 7.3, 5.9, 5.0]
})

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

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

上面的代碼將返回以下內(nèi)容：

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

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

上面的代碼將返回以下內(nèi)容：

14516000

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