[Python] PandasAI

PandasAI는 Pandas의 친숙한 DataFrame 환경 위에 최첨단 생성형 AI(대형 언어 모델, LLM)를 결합한 라이브러리이다. 전통적으로 데이터 분석가는 복잡한 Pandas 메서드 체인이나 긴 SQL 쿼리를 작성해야만 인사이트를 얻을 수 있었다. 그러나 PandasAI를 도입하면, “이번 달 매출 상위 5개 상품을 보여줘” 또는 “지난 3년간 연도별 GDP 합계를 알려줘” 같은 자연어 문장만으로도 즉시 분석을 수행할 수 있다.

귀여웡

주요 특징

• 자연어 질의
  “지난 달 매출 상위 5개 상품”처럼 간단한 문장으로 분석 결과 도출
• 자동 코드 생성
  그룹 연산·통계·시각화 코드를 알아서 만들어 줌
• 간편 통합
  기존 Pandas 코드에 몇 줄만 추가하면 바로 사용 가능
• 다양한 LLM 지원
  OpenAI, Hugging Face 등 원하는 엔진 연결

QA로써 활용할 수 있는 것들은 아래와 같을 것인데, 한번 시작해보지뭐!

 

• 테스트 결과 요약
• 주문·결제 로그 분석
• 품질·성능 지표 시각화
• 이상패턴 탐지

 

 

설치 방법

 

PandasAI는 pip을 사용하여 쉽게 설치할 수 있다. 아래 예제에서는 openai 라이브러리 기반으로 사용할 것이기에 OpenAI 패키지도 함께 설치할 수 있다.

pip install pandasai openai

 

 

또 다른 설치 방법은

 

• PandasAI 팀에서 제공하는 메타 패키지입니다.
• 내부적으로 pandasai 와 openai 를 알맞은 버전으로 함께 설치하도록 의존성을 묶어둔 거라, 한 번에 필요한 것들이 깔립니다.

 

pip install pandasai-openai

 

 

Key 얻으려면, 아래에서 API key 를 얻을 수 있음

https://app.pandabi.ai

PandaBI

 

 

https://docs.pandas-ai.com/v3/large-language-models