CODEWITHSUNDEEP
pythonpandasdataframepandas-groupby
Vaebhav
Vaebhav

Reputation: 5052

Check if Pandas Groupby is empty

I have a usecase , where I m filtering on a row_key which is being supplied as an argument

I m aware of the standard empty property to check for a DataFrame , however could not find a similar one for DataFrameGroupBy

Below is a code snippet that demonstrates the usecase , the data supplied is dummy -

import pandas as pd
from io import StringIO

df = pd.read_csv(StringIO("""
Name,Value,row_key,row_label
XYZ,100,abc,"Label - 2"
ASD,100,abc,"Label - 2"
GHJ,1000,abc,"Label - 2"
KLI,100,abc,"Label - 2"
BHY,1009,bnm,"Label - 2"
TGB,1409,bnm,"Label - 2"
YUJ,1509,bnm,"Label - 2"
KUT,1609,bnm,"Label - 2"
"""))

invalid_row_key = 'fgh'

filter_df = df[df['row_key'] == invalid_row_key].groupby('row_label')

#### I want something similar to below if case , to handle if the filter_df is empty
if filter_df.empty:
   print("No Row Key Present")

I m aware I can use recent_index on the grouped-dataframed , but I wanted to check if there is a better way of handling this

Upvotes: 2

Views: 1699

Answers (2)

Vaebhav
Vaebhav

Reputation: 5052

The best way to handle this would to filter the dataframe before the grouping condition as stated by other answers.

However I tried the following solution myself .


if filter_df.size().empty:
   print("Empty DataFrame")

Upvotes: 0

cs95
cs95

Reputation: 402814

Query the .ngroups attribute, at least 1 group should be present for >= 1 rows:

if not filter_df.ngroups:
    # no row key present

See this post by me for more on ngroups.

The answer above assumes you cannot tell in advance of creating a groupBy object, but if you can, it is better to check at the condition step, then you can check the DataFrame.empty attribute instead which is more intuitive:

filter_df = df[df['row_key'] == invalid_row_key]
if filter_df.empty:
    # no rows present
else:
    filter_df.groupby('row_label').doSomethingElse()

Upvotes: 1

Related Questions