Differences

This shows you the differences between two versions of the page.

--- pandas_groupby [2024/03/26 22:25] – [groupby slicing] raju
+++ pandas_groupby [2024/05/07 20:46] – [extract groupby object by key] raju
@@ Line 235: / Line 235: @@
 Ref: https://stackoverflow.com/questions/49859182/understanding-level-0-and-group-keys
+==== filter elements from groups that dont satisfy a criterion ====
+tags | pandas groupby filter groups
+<code>
+In [2]:
+import pandas as pd
+df = pd.DataFrame({
+    'A' : ['foo', 'bar', 'foo', 'bar', 'foo', 'bar'],
+    'B' : [1, 2, 3, 4, 5, 6],
+    'C' : [2.0, 5., 8., 1., 2., 9.]})
+df
+Out[2]:
+     A  B    C
+  foo  1  2.0
+  bar  2  5.0
+  foo  3  8.0
+  bar  4  1.0
+  foo  5  2.0
+  bar  6  9.0
+In [3]:
+grouped = df.groupby('A')
+In [4]:
+grouped.filter(lambda x: x['B'].mean() > 3.)
+Out[4]:
+     A  B    C
+  bar  2  5.0
+  bar  4  1.0
+  bar  6  9.0
+</code>
 ==== extract groupby object by key ====