Merge branch 'main' of https://github.com/gventuri/pandas-ai

pandasai/helpers/shortcuts.py

@@ -329,5 +329,22 @@ def segment_customers(self, features: list, n_clusters: int) -> DataFrameType:
 Segment customers with the following features and number of clusters:
 features = {features}
 n_clusters = {n_clusters}
+"""
+        )
+
+    def data_summarization(self) -> DataFrameType:
+        """
+        Do summarization of the data.
+
+        Returns:
+            DataFrameType: The summarized DataFrame.
+        """
+
+        return self.chat(
+            """
+Provide the summary in df format
+1. For each distinct value :
+2. Calculate subtotal of rows
+3. Calculate the percentage of the total count for each group
 """
         )