Pivot Table in Pandas Count aggregate()

Question

I am trying to take the count of 'STAGE' occurrence based on project, I used np.size as aggfunc but it return number of occurrence including the project, My count value become double if expected count is 3 means it return 6

I used the below code

df = pd.pivot_table(data_frame, index=['Project'],columns=['Stage'], aggfunc=np.size, fill_value=0)

jezrael · Accepted Answer · 2016-06-30 11:40:52Z

3

You need aggregate function len:

print (data_frame)
  Project Stage
0      an    ip
1     cfc    pe
2      an    ip
3      ap    pe
4     cfc    pe
5      an    ip
6     cfc    ip

df = pd.pivot_table(data_frame, 
                    index='Project',
                    columns='Stage', 
                    aggfunc=len, 
                    fill_value=0)
print (df)
Stage    ip  pe
Project        
an        3   0
ap        0   1
cfc       1   2

Another solution with size:

df = pd.pivot_table(data_frame, 
                    index='Project',
                    columns='Stage', 
                    aggfunc='size', 
                    fill_value=0)
print (df)
Stage    ip  pe
Project        
an        3   0
ap        0   1
cfc       1   2

EDIT by comment:

import matplotlib.pyplot as plt
#all code

df.plot.bar()
plt.show()

edited Jun 30, 2016 at 11:40

answered Jun 30, 2016 at 10:22

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

1 Comment

Vigneshwaran Thenraj Over a year ago

Can we able to convert this pivot table into bar chart directly, Like project as in x - axis and stage as stacked bar

Collectives™ on Stack Overflow

Pivot Table in Pandas Count aggregate()

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related