Pandas

The PandasTools toolkit enables an Agent to perform data manipulation tasks using the Pandas library.

PandasTools enable an Agent to perform data manipulation tasks using the Pandas library.

Prerequisites

The following example requires the pandas and openai libraries.

uv pip install agno -U pandas openai

The Agent model uses an OpenAI key, separately from any toolkit provider credentials.

Set OpenAI Key

Set your OPENAI_API_KEY as an environment variable. You can get one from OpenAI.

export OPENAI_API_KEY=sk-***

Example

cookbook/91_tools/pandas_tools.py
from agno.agent import Agent
from agno.tools.pandas import PandasTools

agent = Agent(
    tools=[PandasTools()],
    description="You are a data analyst with full pandas capabilities for comprehensive data analysis.",
    instructions=[
        "Help users with all aspects of pandas data manipulation",
        "Create, modify, analyze, and visualize DataFrames",
        "Provide detailed explanations of data operations",
        "Suggest best practices for data analysis workflows",
    ],
    markdown=True,
)

agent.print_response("""
Please perform these tasks:
1. Create a pandas dataframe named 'sales_data' using DataFrame() with this sample data:
   {'date': ['2023-01-01', '2023-01-02', '2023-01-03', '2023-01-04', '2023-01-05'],
    'product': ['Widget A', 'Widget B', 'Widget A', 'Widget C', 'Widget B'],
    'quantity': [10, 15, 8, 12, 20],
    'price': [9.99, 15.99, 9.99, 12.99, 15.99]}
2. Show me the first 5 rows of the sales_data dataframe
3. Calculate the total revenue (quantity * price) for each row
""")

Toolkit Params

ParameterTypeDefaultDescription
enable_create_pandas_dataframeboolTrueEnables functionality to create pandas DataFrames.
enable_run_dataframe_operationboolTrueEnables functionality to run DataFrame operations.
allboolFalseEnables all functionality when set to True.

Toolkit Functions

FunctionDescription
create_pandas_dataframeCreates a Pandas DataFrame named dataframe_name by using the specified function create_using_function with parameters function_parameters. Parameters include 'dataframe_name' for the name of the DataFrame, 'create_using_function' for the function to create it (e.g., 'read_csv'), and 'function_parameters' for the arguments required by the function. Returns the name of the created DataFrame if successful, otherwise returns an error message.
run_dataframe_operationRuns a specified operation operation on a DataFrame dataframe_name with the parameters operation_parameters. Parameters include 'dataframe_name' for the DataFrame to operate on, 'operation' for the operation to perform (e.g., 'head', 'tail'), and 'operation_parameters' for the arguments required by the operation. Returns the result of the operation if successful, otherwise returns an error message.

Developer Resources