PandasAI 설치·시작하기 — 데이터에 질문하기

PandasAI는 채팅처럼 데이터에 질문하고 결과를 받아볼 수 있는 파이썬 라이브러리예요. 설치만 하고 LLM을 연결하면 곧바로 데이터 분석을 자연어로 시작할 수 있어요.

설치

PandasAI는 파이썬 3.8+ <=3.11이 필요하고, 의존성 관리는 Poetry를 권장해요.

# poetry 사용 (권장)
poetry add pandasai

# pip 사용
pip install pandasai

LLM 설정

PandasAI를 쓰려면 LLM이 필요해요. 이번 예제에서는 LiteLLM 확장으로 OpenAI 모델을 연결해 볼게요.

pip install pandasai-litellm

import pandasai as pai
from pandasai_litellm.litellm import LiteLLM

llm = LiteLLM(model="gpt-4.1-mini", api_key="YOUR_OPENAI_API_KEY")
pai.config.set({"llm": llm})

데이터와 대화하기

import pandasai as pai
from pandasai_litellm.litellm import LiteLLM

llm = LiteLLM(model="gpt-4.1-mini", api_key="YOUR_OPENAI_API_KEY")
pai.config.set({"llm": llm})

df = pai.read_csv("data/companies.csv")
response = df.chat("What is the average revenue by region?")
print(response)

질문에 따라 반환되는 출력 형식은 다양해요 — 문자열, 데이터프레임, 차트, 숫자 중 하나로 결과가 나와요. 출력 형식에 대한 더 자세한 내용은 여기에서 확인할 수 있어요.

더 알아보기 (Learn more)

출처: PandasAI 공식 문서