2023/09/17

PandasAIを使う(3)

　PandasにChatGPTの機能を実装したPandasAIというライブラリをデータ分析に活用しようと模索しています。先月もスマートウォッチのFitbitのデータを処理するのに利用しました。このPandasaiは日々機能改善されているので、今回Updateしました。

　今まで利用していたライブラリのバージョンは「0.5.4」でした。現在(23/09/17)の最新版は「1.2.2」です。　Update後、Update前に正常に動作していたプログラムを動かすと、以下のエラーが発生。Beautifulsoupをインストールしなさいとのこと。言われた通り、Beautifulsoupをインストールして、改めてプログラムを実行すると、以下の警告が発生。SmartDataframeクラスを利用しなさいとのこと。ネット情報を参考にして、最終的に正常に動いたプログラムは以下の通り。

import pandas as pd
from pandasai import SmartDataframe
from pandasai.llm import OpenAI

# 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
llm = OpenAI(api_token="My key")

df = SmartDataframe(df, config={"llm": llm})
print(df.chat('Which are the 5 happiest countries?'))

結果が正しく表示されました。
　開発がホットな生成AI関連分野のライブラリは、日々の更新をウォッチしておかないとコールする関数が変わっていたりしますね。ChatGPTはともかく、ネット記事でも得られた情報の「新鮮度」を確認しつつ、利用しなくてはなりませんね。