pandas AI

import os
import pandas as pd
from pandasai import Agent

# Sample DataFrame
sales_by_country = pd.DataFrame({
    "country": ["United States", "United Kingdom", "France", 
    "Germany", "Italy", "Spain", "Canada"],

    "revenue": [5000, 3200, 2900, 4100, 2300, 2100, 2500]
})

# By default, unless you choose a different LLM, it will use BambooLLM.
# You can get your free API key signing up
at https://pandabi.ai
os.environ["PANDASAI_API_KEY"] = "YOUR_API_KEY"

agent = Agent(sales_by_country)
agent.chat('Which are the top 5 countries by sales?')