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

cookbook/tools/pandas_tool.py
from phi.agent import Agent
from phi.tools.pandas import PandasTools

# Create an agent with PandasTools
agent = Agent(tools=[PandasTools()])

# Example: Create a dataframe with sample data and get the first 5 rows
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
""")

​
Toolkit Params

ParameterTypeDefaultDescription
dataframesDict[str, pd.DataFrame]{}A dictionary to store Pandas DataFrames, keyed by their names.
create_pandas_dataframefunction-Registers a function to create a Pandas DataFrame.
run_dataframe_operationfunction-Registers a function to run operations on a Pandas DataFrame.

​
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.

​
Information

Was this page helpful?

OpenBBPhi