pandas: Split-apply-combine (2 funcs: count, mean)

ipython-contrib · May 3, 2018 · d0fb866 · d0fb866 · Shi-intel · Apr 27, 2022
1 parent 6af8e5e
commit d0fb866
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/...upyter_contrib_nbextensions/nbextensions/snippets_menu/snippets_submenus_python/pandas.js b/...upyter_contrib_nbextensions/nbextensions/snippets_menu/snippets_submenus_python/pandas.js
@@ -113,7 +113,16 @@ define({
 
         {
             'name' : 'Combining',
-            'snippet' : ['',],
+            'sub-menu' : [
+                {
+                    'name' : 'Split-apply-combine (sum)',
+                    'snippet' : ['df['label_count'] = df.groupby('label', as_index=False)['label'].transform(lambda x: x.count())',],
+                },
+                {
+                    'name' : 'Split-apply-combine (mean)',
+                    'snippet' : ['df['label_mean'] = df.groupby('label', as_index=False)['label'].transform(lambda x: x.mean())',],
+                },
+            ],
         },
 
         {