PandasAI 소개 — 자연어 데이터 대화란 무엇인가

PandasAI는 자연어로 데이터에 질문할 수 있게 해 주는 파이썬 라이브러리예요. 생성형 AI가 사용자의 질문을 이해하고 파이썬 코드나 SQL 쿼리로 바꿔준 뒤, 그 코드로 데이터와 상호작용해 결과를 돌려줘요.

핵심 기능

자연어 쿼리: 데이터에게 자연어로 질문할 수 있어요.
데이터 시각화: 질문 결과를 그래프·차트로 그려줘요.
데이터 정제: 결측값을 처리해 데이터셋을 깨끗하게 만들어요.
특성 생성: 데이터 품질을 높이는 새 특성(feature)을 만들어요.
데이터 커넥터: CSV·XLSX·PostgreSQL·MySQL·BigQuery·Databricks·Snowflake 등 다양한 원천에 연결해요.

어떻게 동작할까

PandasAI는 LLM으로 자연어 질문을 해석해 파이썬 코드나 SQL로 변환하고, 그 코드로 데이터를 조작해 사용자에게 결과를 반환해요. 덕분에 SQL이나 파이썬에 익숙하지 않아도, 혹은 복잡한 코드 없이도 데이터에 질문할 수 있어요.

시작 방법

파이썬 라이브러리로 제공되며 pip나 poetry로 설치해 바로 쓸 수 있어요.

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 공식 문서