Chat and Output Formats

import pandasai as pai

df_customers = pai.read_csv("customers.csv")

response = df_customers.chat("Which are our top 5 customers?")

import pandasai as pai

df_customers = pai.read_csv("customers.csv")
df_orders = pai.read_csv("orders.csv")
df_products = pai.read_csv("products.csv")

response = pai.chat('Who are our top 5 customers and what products do they buy most frequently?', df_customers, df_orders, df_products)

import pandasai as pai

df = pai.read_csv("users.csv")

response = df.chat("Who is the user with the highest age?") # Returns a String response
response = df.chat("How many users in total?") # Returns a Number response
response = df.chat("Show me the data") # Returns a DataFrame response
response = df.chat("Plot the distribution") # Returns a Chart response

response = df.chat("What is the average age?")
print(response)  # Automatically calls response.value
# Output: The average age is 34.5 years

# For charts, printing will display the visualization
chart_response = df.chat("Plot age distribution")
print(chart_response)  # Displays the chart

response = df.chat("Calculate the correlation between age and salary")
print(response.last_code_executed)
# Output: df['age'].corr(df['salary'])

chart_response = df.chat("Create a scatter plot of age vs salary")
chart_response.save("scatter_plot.png")  # Saves the chart as PNG