Pandas+ChatGPT超强组合pandas-ai：交互式数据分析和处理新方法

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，您必须生成自己唯一的 API 密钥。


因为pandas的特性，我们不仅仅可以处理csv文件，我们还可以连接关系型的数据库，例如pgsql：