Python Panda Dataframe Count Specific Values from List

Question

Say I have a list:

mylist = ['a','b','c']

and a Pandas dataframe (df) that has a column named "rating". How can I get the count for number of occurrence of a rating while iterating my list? For example, here is what I need:

for item in myList
   # Do a bunch of stuff in here that takes a long time
   # want to do print statement below to show progress
   # print df['rating'].value_counts().a <- I can do this, 
   #     but want to use variable 'item'
   # print df['rating'].value_counts().item <- Or something like this

I know I can get counts for all distinct values of 'rating', but that is not what I am after.

Chris · Accepted Answer · 2021-09-20 23:33:54Z

1

If you must do it this way, you can use .loc to filter the df prior to getting the size of the resulting df.

mylist = ['a','b','c']
df = pd.DataFrame({'rating':['a','a','b','c','c','c','d','e','f']})


for item in mylist:
    print(item, df.loc[df['rating']==item].size)

Output

a 2
b 1
c 3

answered Sep 20, 2021 at 23:33

Chris

16.3k3 gold badges26 silver badges41 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

ShaneK Over a year ago

Thanks this is what I needed. There is a lot going on behind the hood that I did not include, and my print statement is sort of a progress capture, since each iteration can take a long time. Because of that I did not want to just get everything after my dataframe is complete (I am adding records to it each pass).

Evan W. · Accepted Answer · 2021-09-20 23:39:56Z

1

Instead of thinking about this problem as one of going "from the list to the Dataframe" it might be easiest to flip it around:

mylist = ['a','b','c']
df = pd.DataFrame({'rating':['a','a','b','c','c','c','d','e','f']})

ValueCounts = df['rating'].value_counts()
ValueCounts[ValueCounts.index.isin(mylist)]

Output:

c    3
a    2
b    1
Name: rating, dtype: int64

answered Sep 20, 2021 at 23:39

Evan W.

3422 silver badges10 bronze badges

Comments

U13-Forward · Accepted Answer · 2021-09-20 23:41:29Z

1

You don't even need a for loop, just do:

df['rating'].value_counts()[mylist]

Or to make it a dictionary:

df['rating'].value_counts()[['a', 'b', 'c']].to_dict()

answered Sep 20, 2021 at 23:41

U13-Forward

71.8k15 gold badges100 silver badges125 bronze badges

Collectives™ on Stack Overflow

Python Panda Dataframe Count Specific Values from List

3 Answers 3

1 Comment

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related