PandasAI

PandasAI는 자연어로 데이터에 질문할 수 있게 해주는 Python 라이브러리입니다. 데이터베이스와 데이터 레이크(SQL, CSV, Parquet)를 지원하며, LLM과 RAG를 활용해 데이터 분석을 대화형으로 수행할 수 있습니다.

주요 기능

1. 질문하기: 「지역별 평균 매출은 얼마인가요?」 또는 「판매량 기준 상위 3개 국가의 총매출은 얼마인가요?」와 같이 자연어로 데이터에 질문할 수 있습니다.
2. 차트 시각화: 자연어 프롬프트로 차트를 생성할 수 있습니다. 예를 들어 국가별 GDP를 표시하고 막대마다 다른 색상을 적용한 히스토그램을 만들 수 있습니다.
3. 여러 DataFrame: 여러 DataFrame을 전달하고 서로 연관된 질문을 할 수 있습니다. 예를 들어 직원 데이터와 급여 데이터를 결합해 「가장 많은 급여를 받는 사람은 누구인가요?」라고 물을 수 있습니다.
4. Docker Sandbox: 안전하고 격리된 환경에서 코드를 실행해 악성 공격 위험을 줄입니다. pandasai-docker 패키지를 설치하면 안전한 채팅 상호작용을 위한 DockerSandbox를 초기화할 수 있으며, 사용이 끝나면 샌드박스를 중지해야 합니다.

사용 방법

• Python 요구 사항: 버전 3.8+ <=3.11.
• 설치: 핵심 라이브러리 pandasai와 pandasai-litellm 같은 LLM 통합 패키지를 pip 또는 Poetry로 설치할 수 있습니다.
• 구성 및 실행: LiteLLM을 사용해 gpt-4.1-mini 같은 LLM을 초기화하고, pai.config.set으로 PandasAI를 구성합니다. 그런 다음 pai.read_csv 또는 pai.DataFrame으로 데이터를 불러오고 chat() 메서드나 pai.chat() 함수를 호출합니다.

대상 사용자

데이터와 더 자연스럽게 상호작용하려는 비기술 사용자와 데이터 작업에 드는 시간과 노력을 줄이려는 기술 사용자에게 적합합니다.

주요 장점

복잡한 데이터 질의와 분석을 간단한 자연어 대화로 바꾸고, 여러 데이터 소스 형식을 지원합니다. Docker Sandbox로 코드 실행의 보안을 높이며, 오픈 소스 라이브러리로서 높은 유연성을 제공합니다.

요금 정보

PandasAI 라이브러리는 MIT expat 라이선스로 제공됩니다. 단, pandasai/ee 디렉터리에는 별도의 라이선스가 적용됩니다. 관리형 PandasAI Cloud 또는 셀프 호스팅 Enterprise Offering의 요금은 팀에 문의해야 합니다.