Prerequisites
The following example requires thepandas and openai libraries.
uv pip install -U pandas openai
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
| Parameter | Type | Default | Description |
|---|---|---|---|
enable_create_pandas_dataframe | bool | True | Enables functionality to create pandas DataFrames. |
enable_run_dataframe_operation | bool | True | Enables functionality to run DataFrame operations. |
all | bool | False | Enables all functionality when set to True. |
Toolkit Functions
| Function | Description |
|---|---|
create_pandas_dataframe | Creates 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_operation | Runs 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. |