Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Pandas Count Unique occurrences by Month

Tags:

python

pandas

I have some monthly data that I'm trying to summarize using Pandas and I need to count the number of unique entries that occur each month. Here's some sample code that shows what I'm trying to do:

import pandas as pd

mnths = ['JAN','FEB','MAR','APR']
custs = ['A','B','C',]

testFrame = pd.DataFrame(index=custs, columns=mnths)
testFrame['JAN']['A'] = 'purchased Prod'
testFrame['JAN']['B'] = 'No Data'
testFrame['JAN']['C'] = 'Purchased Competitor'
testFrame['FEB']['A'] = 'purchased Prod'
testFrame['FEB']['B'] = 'purchased Prod'
testFrame['FEB']['C'] = 'purchased Prod'
testFrame['MAR']['A'] = 'No Data'
testFrame['MAR']['B'] = 'No Data'
testFrame['MAR']['C'] = 'Purchased Competitor'
testFrame['APR']['A'] = 'Purchased Competitor'
testFrame['APR']['B'] = 'purchased Prod'
testFrame['APR']['C'] = 'Purchased Competitor'

uniqueValues = pd.Series(testFrame.values.ravel()).unique()

#CODE TO GET COUNT OF ENTRIES IN testFrame BY UNIQUE VALUE

Desired Output:

                JAN    FEB    MAR    APR
purchased Prod   ?     ?       ?      ?
Purchased Competitor ? ?       ?      ?
No Data          ?     ?       ?      ?

I can get the unique values and create a new dataframe with the correct axes/columns

I started here and here: Pandas: Counting unique values in a dataframe Find unique values in a Pandas dataframe, irrespective of row or column location

but still can't quite get the output to the formats I need. I'm not quite sure how to apply the df.groupby syntax or the df.apply syntax to what I'm working with.

like image 686
flyingmeatball Avatar asked May 20 '14 16:05

flyingmeatball


1 Answers

The filling is optional.

In [40]: testFrame.apply(Series.value_counts).fillna(0)
Out[40]: 
                      JAN  FEB  MAR  APR
No Data                 1    0    2    0
Purchased Competitor    1    0    1    2
purchased Prod          1    3    0    1

Here is a neat apply trick. I'll create a function and print out what is incoming (and maybe even debug in their). Then easy to see what's happening.

In [20]: def f(x):
   ....:     print(x)
   ....:     return x.value_counts()
   ....: 

In [21]: testFrame.apply(f)
A          purchased Prod
B                 No Data
C    Purchased Competitor
Name: JAN, dtype: object
A          purchased Prod
B                 No Data
C    Purchased Competitor
Name: JAN, dtype: object
A    purchased Prod
B    purchased Prod
C    purchased Prod
Name: FEB, dtype: object
A                 No Data
B                 No Data
C    Purchased Competitor
Name: MAR, dtype: object
A    Purchased Competitor
B          purchased Prod
C    Purchased Competitor
Name: APR, dtype: object
Out[21]: 
                      JAN  FEB  MAR  APR
No Data                 1  NaN    2  NaN
Purchased Competitor    1  NaN    1    2
purchased Prod          1    3  NaN    1

[3 rows x 4 columns]

So its doing this operation then concatting them together (with the correct labels)

In [22]: testFrame.iloc[0].value_counts()
Out[22]: 
purchased Prod          2
Purchased Competitor    1
No Data                 1
dtype: int64
like image 77
Jeff Avatar answered Sep 23 '22 02:09

Jeff